Skip to content

@PKU-ML PKU-ML

Change the repository type filter

All

    Repositories list

    31 repositories

    • canonicalization

      Public
      Official code for NeurIPS 2024 paper "A Canonicalization Perspective on Invariant and Equivariant Learning".
      eigenvectorscanonicalizationn-bodyinvariancegnnequivarianceequivariant-network
      Python
      •
      MIT License
      •0•1•0•0•Updated Oct 29, 2024Oct 29, 2024

    • PAT

      Public
      Python
      •0•2•0•0•Updated Oct 29, 2024Oct 29, 2024

    • LongPPL

      Public
      Python
      •
      Apache License 2.0
      •0•1•0•0•Updated Oct 28, 2024Oct 28, 2024

    • TERD

      Public
      TERD: A Framework for Backdoor Detection on Diffusion Model
      Python
      •1•6•0•0•Updated Oct 18, 2024Oct 18, 2024

    • ClusterMIM

      Public
      Official Code for ICLR 2024 Paper: On the Role of Discrete Tokenization in Visual Representation Learning
      Python
      •0•3•0•0•Updated Sep 10, 2024Sep 10, 2024

    • TMLlib

      Public
      A Trustworthy Machine Learning Algorithm Library
      Python
      •0•7•0•0•Updated Jul 1, 2024Jul 1, 2024

    • ISSLLib

      Public
      An Interpretable Self-Supervised Learning Library.
      Python
      •0•1•0•0•Updated Jun 28, 2024Jun 28, 2024

    • AdvNotRealFeatures

      Public
      Official Code for reproductivity of the NeurIPS 2023 paper: Adversarial Examples Are Not Real Features
      Python
      •2•14•2•0•Updated Jun 27, 2024Jun 27, 2024

    • Diffusion-PID-Protection

      Public
      Python
      •1•7•0•0•Updated Jun 24, 2024Jun 24, 2024

    • LookAheadLookAround

      Public
      0•1•0•0•Updated Jun 3, 2024Jun 3, 2024

    • nlb

      Public
      Official Code for reproductivity of the ICLR Workshop DPFM 2024 paper: How to Craft Backdoors with Unlabeled Data Alone?
      Python
      •0•3•0•0•Updated May 11, 2024May 11, 2024

    • non_neg

      Public
      Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning
      Python
      •7•36•0•0•Updated Apr 20, 2024Apr 20, 2024

    • adainf

      Public
      Official code for ICLR 2024 paper Do Generated Data Always Help Contrastive Learning?
      Python
      •1•26•2•0•Updated Apr 4, 2024Apr 4, 2024

    • LaplacianCanonization

      Public
      Official code for NeurIPS 2023 paper "Laplacian Canonization: A Minimalist Approach to Sign and Basis Invariant Spectral Embedding".
      spectral-embeddinglaplacian-eigenmapsgnnpositional-encoding
      Python
      •
      MIT License
      •1•7•0•0•Updated Jan 20, 2024Jan 20, 2024

    • Rank-Differential-Mechanism

      Public
      Official code for ICLR 2023 paper "Towards a Unified Theoretical Understanding of Non-contrastive Learning via Rank Differential Mechanism"
      Python
      •
      MIT License
      •1•3•0•1•Updated Nov 15, 2023Nov 15, 2023

    • CLIP-Help-SimCLR

      Public
      Official Code for ICML 2023 Paper: On the Generalization of Multi-modal Contrastive Learning
      Python
      •2•25•0•0•Updated Nov 15, 2023Nov 15, 2023

    • ReBAT

      Public
      Official PyTorch implementation of ReBAT (ReBalanced Adversarial Training) in NeurIPS 2023 "Balance, Imbalance, and Rebalance: Understanding Robust Overfitting from a Minimax Game Perspective".
      Python
      •2•5•0•1•Updated Nov 15, 2023Nov 15, 2023

    • ArchitectureMattersGCL

      Public
      Official code for NeurIPS 2023 "Architecture Matters: Uncovering Implicit Mechanisms in Graph Contrastive Learning"
      Python
      •
      MIT License
      •1•8•0•0•Updated Nov 10, 2023Nov 10, 2023

    • Tri-factor-Contrastive-Learning

      Public
      Official Code for NeurIPS 2023 Paper: Identifiable Contrastive Learning with Automatic Feature Importance Discovery
      Python
      •0•4•1•0•Updated Oct 31, 2023Oct 31, 2023

    • Generalist

      Public
      CVPR 2023 generalist
      Python
      •
      MIT License
      •3•15•0•0•Updated Oct 25, 2023Oct 25, 2023

    • CFA

      Public
      Python
      •3•17•1•0•Updated Sep 1, 2023Sep 1, 2023

    • ContraNorm

      Public
      Official code for ICLR 2023 paper "ContraNorm: A Contrastive Learning Perspective on Oversmoothing and Beyond "
      Python
      •
      MIT License
      •4•32•0•0•Updated Apr 24, 2023Apr 24, 2023

    • Message-Passing-Contrastive-Learning

      Public
      Official Code for ICLR 2023 Paper: A Message Passing Perspective on Learning Dynamics of Contrastive Learning
      Python
      •0•10•0•0•Updated Mar 9, 2023Mar 9, 2023

    • DYNACL

      Public
      [ICLR 2023] Official repository of the paper "Rethinking the Effect of Data Augmentation in Adversarial Contrastive Learning"
      Python
      •2•17•1•0•Updated Feb 19, 2023Feb 19, 2023

    • AT-for-OOD

      Public
      Official code for NeurIPS 2022 paper "Improving Out-of-Distribution Generalization by Adversarial Training with Structured Priors"
      Python
      •0•3•0•0•Updated Feb 6, 2023Feb 6, 2023

    • When-Adversarial-Training-Meets-Vision-Transformers

      Public
      Official implementation of "When Adversarial Training Meets Vision Transformers: Recipes from Training to Architecture" published at NeurIPS 2022.
      Python
      •4•1•0•0•Updated Jan 29, 2023Jan 29, 2023

    • U-MAE

      Public
      Official Code for NeurIPS 2022 Paper: How Mask Matters: Towards Theoretical Understandings of Masked Autoencoders
      Python
      •5•1•0•0•Updated Oct 19, 2022Oct 19, 2022

    • ARC

      Public
      Official Code for ICLR2022 Paper: Chaos is a Ladder: A New Theoretical Understanding of Contrastive Learning via Augmentation Overlap
      Jupyter Notebook
      •3•0•0•0•Updated Sep 2, 2022Sep 2, 2022

    • Self-Ensemble-Adversarial-Training

      Public
      SEAT
      Python
      •3•0•0•0•Updated Mar 23, 2022Mar 23, 2022

    • DGC

      Public
      Official Code of Decoupled Graph Convolution (DGC)
      Python
      •
      MIT License
      •4•0•0•0•Updated Nov 27, 2021Nov 27, 2021