OPTML Group

All

39 repositories

OPTML-Group.github.io
Public
SCSS
•3•1•0•0•Updated Feb 22, 2026Feb 22, 2026
Unlearn_Optimizer
Public
Python
•
MIT License
•0•2•0•0•Updated Feb 9, 2026Feb 9, 2026
ZO-Muon
Public
Python
•
MIT License
•0•0•0•0•Updated Feb 9, 2026Feb 9, 2026
MemorizationBarrier
Public
This is the code for paper: Breaking Memorization Barriers in LLM Code Fine-Tuning via Information Bottleneck for Improved Generalization
Python
•
Apache License 2.0
•0•1•0•0•Updated Dec 31, 2025Dec 31, 2025
Unlearn-Backdoor
Public
Python
•
GNU General Public License v3.0
•0•4•0•0•Updated Dec 28, 2025Dec 28, 2025
Unlearn-R2MU
Public
[EMNLP 2025] Reasoning Model Unlearning: Forgetting Traces, Not Just Answers, While Preserving Reasoning Skills
Python
•0•5•0•0•Updated Nov 3, 2025Nov 3, 2025
Unlearn-FullStack
Public
Python
•
MIT License
•0•2•0•0•Updated Oct 12, 2025Oct 12, 2025
Unlearn-Simple
Public
[NeurIPS25] Official repo for "Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning"
data-privacy language-model machine-unlearning trustworthy-machine-learning trustworthy-ai large-language-models llm-unlearning
Python
•
MIT License
•12•42•1•1•Updated Oct 3, 2025Oct 3, 2025
Unlearn-Smooth
Public
[ICML25] Official repo for "Towards LLM Unlearning Resilient to Relearning Attacks: A Sharpness-Aware Minimization Perspective and Beyond"
Python
•
MIT License
•0•20•0•0•Updated Sep 27, 2025Sep 27, 2025
Unlearn-Trace
Public
Unlearning Isn't Invisible: Detecting Unlearning Traces in LLMs from Model Outputs
machine-learning machine-unlearning trustworthy-ai llm large-language-model
Python
•
MIT License
•1•24•0•0•Updated Jul 5, 2025Jul 5, 2025
CyclicReflex
Public
"CyclicReflex: Improving Large Reasoning Models via Cyclical Reflection Token Scheduling" by Chongyu Fan, Yihua Zhang, Jinghan Jia, Alfred Hero, Sijia Liu
Python
•
MIT License
•0•5•1•0•Updated Jun 22, 2025Jun 22, 2025
VLM-Safety-Unlearn
Public
Safety Mirage: How Spurious Correlations Undermine VLM Safety Fine-tuning
machine-unlearning trustworthy-machine-learning vision-language-model multimodal-large-language-models safety-alignment
Python
•
MIT License
•0•14•1•0•Updated Jun 17, 2025Jun 17, 2025
Unlearn-ILU
Public
Python
•
MIT License
•1•6•0•0•Updated Jun 15, 2025Jun 15, 2025
EPiC
Public
Python
•0•4•0•0•Updated Jun 11, 2025Jun 11, 2025
Unlearn-Saliency
Public
[ICLR24 (Spotlight)] "SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation" by Chongyu Fan*, Jian…
generative-model data-privacy diffusion machine-unlearning forgetting diffusion-models data-deletion membership-inference-attack unlearning membership-inference
Python
•
MIT License
•29•141•4•0•Updated May 27, 2025May 27, 2025
Unlearn-WorstCase
Public
[ECCV24] "Challenging Forgets: Unveiling the Worst-Case Forget Sets in Machine Unlearning" by Chongyu Fan*, Jiancheng Liu*, Alfred Hero, Sijia Liu
evaluation data-privacy evaluation-framework machine-unlearning forgetting data-deletion unlearning data-removal
Python
•
MIT License
•2•25•2•0•Updated May 27, 2025May 27, 2025
MU-Coreset
Public
[COLM2025]"LLM Unlearning Reveals a Stronger-Than-Expected Coreset Effect in Current Benchmarks"
Python
•1•2•1•0•Updated Apr 22, 2025Apr 22, 2025
Diffusion-MU-Attack
Public
The official implementation of ECCV'24 paper "To Generate or Not? Safety-Driven Unlearned Diffusion Models Are Still Easy To Generate Unsafe Images ... For Now"…
evaluation-framework robustness adversarial-attacks unlearning stable-diffusion attack-unlearned-diffusion-model
Python
•
MIT License
•5•87•4•0•Updated Feb 28, 2025Feb 28, 2025
WAGLE
Public
Official repo for NeurIPS'24 paper "WAGLE: Strategic Weight Attribution for Effective and Modular Unlearning in Large Language Models"
Python
•
MIT License
•3•18•1•0•Updated Dec 16, 2024Dec 16, 2024
UnlearnCanvas
Public
[NeurIPS 2024 D&B Track] UnlearnCanvas: A Stylized Image Dataset to Benchmark Machine Unlearning for Diffusion Models by Yihua Zhang, Chongyu Fan, Yimeng Zhang,…
Python
•5•82•15•0•Updated Nov 11, 2024Nov 11, 2024
AdvUnlearn
Public
Official implementation of NeurIPS'24 paper "Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models". This work adversari…
robust-optimization adversarial-machine-learning unlearning stable-diffusion unlearned-diffusion-model
Jupyter Notebook
•
Creative Commons Attribution 4.0 International
•3•49•1•0•Updated Nov 4, 2024Nov 4, 2024
DeepZero
Public
[ICLR'24] "DeepZero: Scaling up Zeroth-Order Optimization for Deep Model Training" by Aochuan Chen*, Yimeng Zhang*, Jinghan Jia, James Diffenderfer, Jiancheng L…
blackbox-optimization zeroth-order-optimization efficient-deep-learning ai4science
Python
•
MIT License
•10•70•2•0•Updated Oct 9, 2024Oct 9, 2024
SOUL
Public
Official repo for EMNLP'24 paper "SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning"
Python
•
MIT License
•6•29•1•0•Updated Oct 1, 2024Oct 1, 2024
QF-Attack
Public
[CVPR23W] "A Pilot Study of Query-Free Adversarial Attack against Stable Diffusion" by Haomin Zhuang, Yihua Zhang and Sijia Liu
Python
•2•26•0•0•Updated Aug 27, 2024Aug 27, 2024
BiBadDiff
Public
"From Trojan Horses to Castle Walls: Unveiling Bilateral Backdoor Effects in Diffusion Models" by Zhuoshi Pan*, Yuguang Yao*, Gaowen Liu, Bingquan Shen, H. Vick…
Python
•2•7•1•0•Updated Mar 25, 2024Mar 25, 2024
BackdoorMSPC
Public
[ICLR2024]"Backdoor Secrets Unveiled: Identifying Backdoor Data with Optimized Scaled Prediction Consistency" by Soumyadeep Pal, Yuguang Yao, Ren Wang, Bingquan…
Python
•0•4•1•0•Updated Mar 14, 2024Mar 14, 2024
Unlearn-Sparse
Public
[NeurIPS23 (Spotlight)] "Model Sparsity Can Simplify Machine Unlearning" by Jinghan Jia*, Jiancheng Liu*, Parikshit Ram, Yuguang Yao, Gaowen Liu, Yang Liu, Pran…
data-privacy machine-unlearning forgetting data-deletion membership-inference-attack unlearning membership-inference data-removal
Python
•
MIT License
•11•83•3•0•Updated Mar 12, 2024Mar 12, 2024
.github
Public
0•0•0•0•Updated Feb 11, 2024Feb 11, 2024
BLO-Toolbox
Public
1•1•0•0•Updated Dec 18, 2023Dec 18, 2023
DP4TL
Public
[NeurIPS2023] "Selectivity Drives Productivity: Efficient Dataset Pruning for Enhanced Transfer Learning" by Yihua Zhang*, Yimeng Zhang*, Aochuan Chen*, Jinghan…
Python
•2•14•1•0•Updated Oct 12, 2023Oct 12, 2023