SDS-Lab / ROT-PoolingLinks
Learnable Global Pooling Layers Based on Regularized Optimal Transport (ROT)
☆16Updated last year
Alternatives and similar repositories for ROT-Pooling
Users that are interested in ROT-Pooling are comparing it to the libraries listed below
Sorting:
- A Quasi-Wasserstein Loss for Learning Graph Neural Networks (QW loss)☆10Updated last year
- [ICML2025] The official implementation of "PolyConf: Unlocking Polymer Conformation Generation through Hierarchical Generative Models"☆18Updated 3 weeks ago
- Official implementation for Learning Invariant Molecular Representation in Latent Discrete Space (NeurIPS 2023)☆21Updated last year
- A list of papers for group meeting☆16Updated last month
- [ICML2025] The official implementation of "WGFormer: An SE(3)-Transformer Driven by Wasserstein Gradient Flows for Molecular Ground-State…☆17Updated 3 weeks ago
- [AAAI2023 Oral] The official implementation of "Hierarchical Contrastive Learning for Temporal Point Processes"☆25Updated 2 months ago
- [NeurIPS'23] Source code of "Data-Centric Learning from Unlabeled Graphs with Diffusion Model": A data-centric transfer learning framewor…☆21Updated 3 weeks ago
- [NeurIPS 2023] "Rethinking Tokenizer and Decoder in Masked Graph Modeling for Molecules"☆37Updated last year
- [CIKM2023] The official implementation of "MPerformer: An SE(3) Transformer-based Molecular Perceptron"☆20Updated 7 months ago
- Part of official implementation of "Natural language-informed learning of molecule graphs"☆18Updated last year
- [ICLR 2023] Learnable Randomness Injection (LRI) for interpretable Geometric Deep Learning.☆23Updated last year
- [EMNLP 2023] ReLM: Leveraging Language Models for Enhanced Chemical Reaction Prediction.☆21Updated last year
- [ICLR 2023] "Graph Domain Adaptation via Theory-Grounded Spectral Regularization" by Yuning You, Tianlong Chen, Zhangyang Wang, Yang Shen☆21Updated 2 years ago
- [NeurIPS 2023] Does Invariant Graph Learning via Environment Augmentation Learn Invariance?☆21Updated 10 months ago
- The code of “Prototypical Graph Contrastive Learning”. [TNNLS 2022]☆24Updated 2 years ago
- Source code for the paper 'Uncovering Neural Scaling Laws in Molecular Representation Learning' (NeurIPS 2023 Datasets and Benchmarks).☆14Updated last year
- The official source code for "Conditional Graph Information Bottleneck for Molecular Relational Learning".☆41Updated 2 years ago
- ☆16Updated last year
- Ratioanle-aware Graph Contrastive Learning codebase☆43Updated 2 years ago
- ☆24Updated 2 years ago
- [ICML 2022] pGNN, p-Laplacian Based Graph Neural Networks☆27Updated 2 years ago
- NeurIPS'22 Oral: EquiVSet - Learning Neural Set Functions Under the Optimal Subset Oracle☆19Updated 2 years ago
- Code for AAAI24 paper Text-Guided Molecule Generation with Diffusion Language Model☆25Updated 10 months ago
- The code for GIMLET: A Unified Graph-Text Model for Instruction-Based Molecule Zero-Shot Learning☆62Updated last year
- [KDD 2022] Implementation of "Feature Overcorrelation in Deep Graph Neural Networks: A New Perspective"☆45Updated last year
- InstructMol: Multi-Modal Integration for Building a Versatile and Reliable Molecular Assistant in Drug Discovery (COLING 2025)☆47Updated 6 months ago
- Official implementation for the paper "Learning Substructure Invariance for Out-of-Distribution Molecular Representations" (NeurIPS 2022)…☆61Updated 2 years ago
- Pytorch implementation for ICLR24:"Online GNN Evaluation Under Test-Time Graph Distribution Shifts"☆16Updated last year
- [ECCV'22] Equivariant Hypergraph Neural Networks, in PyTorch☆30Updated 2 years ago
- ☆14Updated 3 years ago