Skip to content

@PKU-ML PKU-ML

Change the repository type filter

All

    Repositories list

    43 repositories

    • STS

      Public
      Official Code for ICLR 2026 Paper: SAE as a Crystal Ball: Interpretable Features Predict Cross-domain Transferability of LLMs without Training
      Python
      •0•3•0•0•Updated Mar 3, 2026Mar 3, 2026

    • Any-order-Any-subset-AR

      Public
      0•1•0•0•Updated Mar 1, 2026Mar 1, 2026

    • CoT-Length

      Public
      Python
      •0•1•0•0•Updated Feb 28, 2026Feb 28, 2026

    • 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
      •1•2•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
      •11•110•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•40•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•21•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
      •3•10•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-body
      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•12•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
      •8•47•2•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-eigenmapsgnn
      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 …
      Python
      •1•7•0•1•Updated Nov 15, 2023Nov 15, 2023
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.