yusx-swapp / GNN-RL-Model-CompressionLinks
GNN-RL Compression: Topology-Aware Network Pruning using Multi-stage Graph Embedding and Reinforcement Learning
☆68Updated 2 years ago
Alternatives and similar repositories for GNN-RL-Model-Compression
Users that are interested in GNN-RL-Model-Compression are comparing it to the libraries listed below
Sorting:
- Link Scheduling using Graph Neural Networks, IEEE TWC☆35Updated last year
- This project will realize experiments about BranchyNet partitioning using pytorch framework☆27Updated 4 years ago
- Implementation of our NeurIPS 2021 paper "A Bi-Level Framework for Learning to Solve Combinatorial Optimization on Graphs".☆100Updated 2 years ago
- ☆22Updated 2 years ago
- [IEEE Access] "Head Network Distillation: Splitting Distilled Deep Neural Networks for Resource-constrained Edge Computing Systems" and […☆36Updated last year
- Measuring and predicting on-device metrics (latency, power, etc.) of machine learning models☆66Updated 2 years ago
- PipeEdge: Pipeline Parallelism for Large-Scale Model Inference on Heterogeneous Edge Devices☆33Updated last year
- This collection of papers can be used to summarize research about graph reinforcement learning for the convenience of researchers.☆182Updated 6 months ago
- Reference code for https://arxiv.org/abs/1906.08879☆16Updated 5 years ago
- Official Repo for the NeurIPS2020 paper "Erdos Goes Neural: An Unsupervised Learning Framework for Combinatorial Optimization on Graphs"☆49Updated 3 years ago
- The code of paper Learning Cut Selection for Mixed-Integer Linear Programming via Hierarchical Sequence Model. Zhihai Wang, Xijun Li,…☆57Updated 2 years ago
- Pytorch-based early exit network inspired by branchynet☆31Updated 3 weeks ago
- Code of NeurIPS paper: arxiv.org/abs/2302.08224☆202Updated 8 months ago
- PyTorch implementation of the paper: Multi-Agent Collaborative Inference via DNN Decoupling: Intermediate Feature Compression and Edge Le…☆40Updated last year
- ☆229Updated 2 years ago
- ☆92Updated 2 years ago
- A PyTorch Implementation for experiements in paper: Neurosurgeon: Collaborative Intelligence Between the Cloud and Mobile Edge.☆13Updated 2 years ago
- Autodidactic Neurosurgeon Collaborative Deep Inference for Mobile Edge Intelligence via Online Learning☆40Updated 3 years ago
- ☆13Updated 2 years ago
- [ICML 2021] "A Unified Lottery Tickets Hypothesis for Graph Neural Networks", Tianlong Chen*, Yongduo Sui*, Xuxi Chen, Aston Zhang, Zhang…☆66Updated last year
- Distributed deep learning cluster simulation environment and RL-GNN resource management implementations.☆14Updated 2 years ago
- Paper collection of reinforcement learning based combinatorial optimization☆54Updated 4 years ago
- Deep reinforcement learning for REsource Allocation in streaM processing☆30Updated 2 years ago
- FedFormer: Contextual Federation with Attention in Reinforcement Learning (AAMAS 2023)☆43Updated 6 months ago
- Hao Jin, Yang Peng, Wenhao Yang, Shusen Wang and Zhihua Zhang. Federated Reinforcement Learning with Environment Heterogeneity. AISTATS, …☆58Updated 3 years ago
- Attention based model for learning to solve different routing problems☆40Updated 2 years ago
- Pytorch Implementation of Neural Architecture Search with Reinforcement Learning (in dev)☆18Updated 5 years ago
- ☆36Updated 2 years ago
- This open source library is available to summarize several years of research papers on graph reinforcement learning for the convenience o…☆20Updated 3 years ago
- ☆40Updated 4 years ago