Skip to content
@OPTML-Group

OPTML Group

Welcome to the OPTML Group's GitHub Repository!

About Us

OPtimization and Trustworthy Machine Learning (OPTML) group (Group Website) is an active research group at Michigan State University. Our research interests span the areas of machine learning (ML)/deep learning (DL), optimization, computer vision, security, signal processing and data science, with a focus on developing learning algorithms and theory, as well as robust and explainable artificial intelligence (AI). These research themes provide a solid foundation for reaching the long-term research objective: Making AI systems scalable and trustworthy.

As AI moves from the lab into the real world (e.g., autonomous vehicles), ensuring its safety becomes a paramount requirement prior to its deployment. Moreover, as datasets, ML/DL models, and learning tasks become increasingly complex, getting ML/DL to scale calls for new advances in learning algorithm design. More broadly, the study towards robust and scalable AI could make a significant impact on machine learning theories, and induce more promising applications in, e.g., automated ML, meta-learning, privacy and security, hardware design, and big data analysis. We seek a new learning frontier when the current learning algorithms become infeasible, and formalize foundations of secure learning.

We always look for passionate students to join the team in terms of RA/TA/externship/internship/visiting students (more info)!

Pinned Loading

  1. Unlearn-Saliency Unlearn-Saliency Public

    [ICLR24 (Spotlight)] "SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation" by Chongyu Fan*, Jiancheng Liu*, Yihua Zhang, Eric Wong, D…

    Python 108 18

  2. UnlearnCanvas 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, Yuguang Yao, Jinghan Jia, Jiancheng …

    Python 59 2

  3. Diffusion-MU-Attack 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". This work introduces one fast and e…

    Python 60 3

  4. AdvUnlearn AdvUnlearn Public

    Official implementation of NeurIPS'24 paper "Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models". This work adversarially unlearns the text encoder to enh…

    Jupyter Notebook 31 1

  5. Unlearn-Sparse Unlearn-Sparse Public

    [NeurIPS23 (Spotlight)] "Model Sparsity Can Simplify Machine Unlearning" by Jinghan Jia*, Jiancheng Liu*, Parikshit Ram, Yuguang Yao, Gaowen Liu, Yang Liu, Pranay Sharma, Sijia Liu

    Python 65 7

  6. Unlearn-Simple Unlearn-Simple Public

    "Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning" by Chongyu Fan*, Jiancheng Liu*, Licong Lin*, Jinghan Jia, Ruiqi Zhang, Song Mei, Sijia Liu

    Python 18 6

Repositories

Showing 10 of 27 repositories
  • OPTML-Group/OPTML-Group.github.io’s past year of commit activity
    SCSS 1 2 0 0 Updated Dec 22, 2024
  • WAGLE Public

    Official repo for NeurIPS'24 paper "WAGLE: Strategic Weight Attribution for Effective and Modular Unlearning in Large Language Models"

    OPTML-Group/WAGLE’s past year of commit activity
    Python 10 MIT 3 0 0 Updated Dec 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, Yuguang Yao, Jinghan Jia, Jiancheng Liu, Gaoyuan Zhang, Gaowen Liu, Ramana Kompella, Xiaoming Liu, Sijia Liu

    OPTML-Group/UnlearnCanvas’s past year of commit activity
    Python 59 2 8 0 Updated Nov 11, 2024
  • Unlearn-Simple Public

    "Simplicity Prevails: Rethinking Negative Preference Optimization for LLM Unlearning" by Chongyu Fan*, Jiancheng Liu*, Licong Lin*, Jinghan Jia, Ruiqi Zhang, Song Mei, Sijia Liu

    OPTML-Group/Unlearn-Simple’s past year of commit activity
    Python 18 MIT 6 0 0 Updated Nov 6, 2024
  • Unlearn-Saliency Public

    [ICLR24 (Spotlight)] "SalUn: Empowering Machine Unlearning via Gradient-based Weight Saliency in Both Image Classification and Generation" by Chongyu Fan*, Jiancheng Liu*, Yihua Zhang, Eric Wong, Dennis Wei, Sijia Liu

    OPTML-Group/Unlearn-Saliency’s past year of commit activity
    Python 108 MIT 18 0 0 Updated Nov 4, 2024
  • 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". This work introduces one fast and effective attack method to evaluate the harmful-content generation ability of safety-driven unlearned diffusion models.

    OPTML-Group/Diffusion-MU-Attack’s past year of commit activity
    Python 60 MIT 3 2 1 Updated Nov 4, 2024
  • AdvUnlearn Public

    Official implementation of NeurIPS'24 paper "Defensive Unlearning with Adversarial Training for Robust Concept Erasure in Diffusion Models". This work adversarially unlearns the text encoder to enhance the robustness of unlearned DMs against adversarial prompt attacks and achieves a better balance between unlearning performance and image generation

    OPTML-Group/AdvUnlearn’s past year of commit activity
    Jupyter Notebook 31 CC-BY-4.0 1 1 0 Updated Nov 4, 2024
  • Unlearn-WorstCase Public

    [ECCV24] "Challenging Forgets: Unveiling the Worst-Case Forget Sets in Machine Unlearning" by Chongyu Fan*, Jiancheng Liu*, Alfred Hero, Sijia Liu

    OPTML-Group/Unlearn-WorstCase’s past year of commit activity
    Python 19 MIT 1 0 0 Updated Oct 9, 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 Liu, Konstantinos Parasyris, Yihua Zhang, Zheng Zhang, Bhavya Kailkhura, Sijia Liu

    OPTML-Group/DeepZero’s past year of commit activity
    Python 47 MIT 6 1 0 Updated Oct 9, 2024
  • SOUL Public

    Official repo for EMNLP'24 paper "SOUL: Unlocking the Power of Second-Order Optimization for LLM Unlearning"

    OPTML-Group/SOUL’s past year of commit activity
    Python 15 MIT 2 1 0 Updated Oct 1, 2024

People

This organization has no public members. You must be a member to see who’s a part of this organization.

Top languages

Loading…

Most used topics

Loading…