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"☆20Updated 2 months ago
- [NeurIPS 2023] "Rethinking Tokenizer and Decoder in Masked Graph Modeling for Molecules"☆38Updated last year
- Official implementation for Learning Invariant Molecular Representation in Latent Discrete Space (NeurIPS 2023)☆21Updated last year
- [ICML2025] The official implementation of "WGFormer: An SE(3)-Transformer Driven by Wasserstein Gradient Flows for Molecular Ground-State…☆17Updated 2 months ago
- [CIKM2023] The official implementation of "MPerformer: An SE(3) Transformer-based Molecular Perceptron"☆20Updated 8 months ago
- [ICLR 2023] Learnable Randomness Injection (LRI) for interpretable Geometric Deep Learning.☆23Updated 2 years ago
- [EMNLP 2023] ReLM: Leveraging Language Models for Enhanced Chemical Reaction Prediction.☆21Updated last year
- [AAAI2023 Oral] The official implementation of "Hierarchical Contrastive Learning for Temporal Point Processes"☆26Updated 4 months ago
- [NeurIPS'23] Source code of "Data-Centric Learning from Unlabeled Graphs with Diffusion Model": A data-centric transfer learning framewor…☆21Updated 2 months ago
- Part of official implementation of "Natural language-informed learning of molecule graphs"☆18Updated 2 years ago
- ☆24Updated 2 years ago
- The code for GIMLET: A Unified Graph-Text Model for Instruction-Based Molecule Zero-Shot Learning☆62Updated last year
- [KDD'22] Source codes of "Graph Rationalization with Environment-based Augmentations"☆45Updated 4 months ago
- Official implementation for the paper "Learning Substructure Invariance for Out-of-Distribution Molecular Representations" (NeurIPS 2022)…☆61Updated 2 years ago
- InstructMol: Multi-Modal Integration for Building a Versatile and Reliable Molecular Assistant in Drug Discovery (COLING 2025)☆47Updated 8 months ago
- A list of papers for group meeting☆16Updated 2 months ago
- Code for the papers: "Graph Representation Learning for Multi-Task Settings: a Meta-Learning Approach", "A Meta-Learning Approach for Gra…☆18Updated 3 years ago
- Edge-Augmented Graph Transformer☆78Updated last year
- Implementation of Self-supervised Graph-level Representation Learning with Local and Global Structure (ICML 2021).☆80Updated 4 years ago
- Implementation for the paper MoCL: Contrastive Learning on Molecular Graph with multi-level Domain Knowledge☆41Updated last year
- The official source code for "Conditional Graph Information Bottleneck for Molecular Relational Learning".☆43Updated 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 PyTorch implementation of MoMu, described in "Natural Language-informed Modeling of Molecule Graphs".☆29Updated 2 years ago
- Conditional Diffusion Based on Discrete Graph Structures for Molecular Graph Generation☆34Updated 2 years ago
- [IJCAI 2023] Graph Propagation Transformer for Graph Representation Learning☆49Updated last year
- Source Code of IJCAI 2022 paper "Fine-Tuning Graph Neural Networks via Graph Topology induced Optimal Transport"☆24Updated 2 years ago
- [CIKM2024] The official implementation of "MMPolymer: A Multimodal Multitask Pretraining Framework for Polymer Property Prediction"☆26Updated last week
- ☆13Updated 2 years ago
- Code of our IJCAI2021 paper: "Learning Attributed Graph Representation with Communicative Message Passing Transformer"☆40Updated 3 years ago