Skip to content

@PKU-ML PKU-ML

Change the repository type filter

All

    Repositories list

    40 repositories

    • SSL4RL

      Public
      The official implementations of the Paper "SSL4RL: Revisiting Self-supervised Learning as Intrinsic Reward for Visual-Language Reasoning"
      Python
      •
      Apache License 2.0
      •0•0•0•0•Updated Dec 23, 2025Dec 23, 2025

    • Cross-Class-Features-AT

      Public
      Python
      •0•1•0•0•Updated Dec 18, 2025Dec 18, 2025

    • LongPPL

      Public
      Code for ICLR 2025 Paper "What is Wrong with Perplexity for Long-context Language Modeling?"
      Python
      •
      Apache License 2.0
      •8•109•0•0•Updated Oct 11, 2025Oct 11, 2025

    • LLDMs-FDM

      Public
      Official code for the foreseeing decoding method for the decoding of LLDMs.
      0•1•0•0•Updated Oct 2, 2025Oct 2, 2025

    • GARC

      Public
      Code for JMLR Paper "An Augmentation Overlap Theory of Contrastive Learning"
      Jupyter Notebook
      •0•1•0•0•Updated Sep 30, 2025Sep 30, 2025

    • G1

      Public
      Python
      •
      Apache License 2.0
      •3•34•3•0•Updated Jun 14, 2025Jun 14, 2025

    • LongShortAlignment

      Public
      0•1•1•0•Updated Jun 4, 2025Jun 4, 2025

    • PAT

      Public
      Code for NeurIPS 2024 Paper "Fight Back Against Jailbreaking via Prompt Adversarial Tuning"
      Python
      •3•22•0•0•Updated May 6, 2025May 6, 2025

    • CFA

      Public
      Python
      •2•20•1•0•Updated Mar 14, 2025Mar 14, 2025

    • Projector_Theory

      Public
      Code for ICLR 2025 Paper "Projection Head is Secretly an Information Bottleneck"
      Python
      •0•4•0•0•Updated Mar 12, 2025Mar 12, 2025

    • InvICL

      Public
      Code for ICLR 2025 Paper "Rethinking Invariance in In-context Learning"
      0•1•1•0•Updated Mar 1, 2025Mar 1, 2025

    • Diffusion-PID-Protection

      Public
      Code for ICML 2024 Paper "PID: Prompt-Independent Data Protection Against Latent Diffusion Models"
      Python
      •1•12•0•0•Updated Jan 27, 2025Jan 27, 2025

    • Monosemanticity-Robustness

      Public
      Official Code for ICLR 2025 Paper "Beyond Interpretability: The Gains of Feature Monosemanticity on Model Robustness"
      Python
      •2•8•0•0•Updated Nov 2, 2024Nov 2, 2024

    • canonicalization

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

    • TERD

      Public
      Code for ICML 2024 Paper "TERD: A Framework for Backdoor Detection on Diffusion Model"
      Python
      •2•9•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•7•0•0•Updated Sep 10, 2024Sep 10, 2024

    • TMLlib

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

    • ISSLLib

      Public
      An Interpretable Self-Supervised Learning Library.
      Python
      •0•4•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•16•2•0•Updated Jun 27, 2024Jun 27, 2024

    • LookAheadLookAround

      Public
      0•2•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•5•0•0•Updated May 11, 2024May 11, 2024

    • non_neg

      Public
      Official Code for ICLR 2024 Paper: Non-negative Contrastive Learning
      Python
      •7•46•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•31•4•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•5•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•26•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
      •1•7•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•9•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•5•1•0•Updated Oct 31, 2023Oct 31, 2023

    • Generalist

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