yusx-swapp / GNN-RL-Model-Compression
GNN-RL Compression: Topology-Aware Network Pruning using Multi-stage Graph Embedding and Reinforcement Learning
☆56Updated last year
Related projects ⓘ
Alternatives and complementary repositories for GNN-RL-Model-Compression
- Link Scheduling using Graph Neural Networks, IEEE TWC☆29Updated 8 months ago
- This collection of papers can be used to summarize research about graph reinforcement learning for the convenience of researchers.☆152Updated 2 months ago
- [IEEE Access] "Head Network Distillation: Splitting Distilled Deep Neural Networks for Resource-constrained Edge Computing Systems" and […☆35Updated last year
- ☆12Updated last year
- Autodidactic Neurosurgeon Collaborative Deep Inference for Mobile Edge Intelligence via Online Learning☆37Updated 3 years ago
- Pytorch Implementation of Neural Architecture Search with Reinforcement Learning (in dev)☆19Updated 4 years ago
- Measuring and predicting on-device metrics (latency, power, etc.) of machine learning models☆66Updated last year
- [ICML 2021] "A Unified Lottery Tickets Hypothesis for Graph Neural Networks", Tianlong Chen*, Yongduo Sui*, Xuxi Chen, Aston Zhang, Zhang…☆63Updated last year
- ☆88Updated last year
- ☆20Updated last year
- ☆41Updated 2 years ago
- Using Feature Decomposition method to accelerate GNN inference☆12Updated 3 years ago
- 2-stage pruning to favor distributed inference (local device compute half of the model, upload the feature for further computing on stron…☆23Updated 6 years ago
- ☆11Updated 5 years ago
- 2021 Summer Research Internship project (UROP) at Imperial College London. Supervised by Prof George Constantinides and Ben Biggs☆16Updated last year
- This project will realize experiments about BranchyNet partitioning using pytorch framework☆28Updated 4 years ago
- Implementation of our NeurIPS 2021 paper "A Bi-Level Framework for Learning to Solve Combinatorial Optimization on Graphs".☆93Updated last year
- Official Repo for the NeurIPS2020 paper "Erdos Goes Neural: An Unsupervised Learning Framework for Combinatorial Optimization on Graphs"☆44Updated 2 years ago
- HEFT, randomHEFT and IPEFT algorithms for static list DAG Scheduling☆48Updated 5 years ago
- A PyTorch Implementation for experiements in paper: Neurosurgeon: Collaborative Intelligence Between the Cloud and Mobile Edge.☆10Updated last year
- Deep Compressive Offloading: Speeding Up Neural Network Inference by Trading Edge Computation for Network Latency☆25Updated 3 years ago
- This open source library is available to summarize several years of research papers on graph reinforcement learning for the convenience o…☆19Updated 2 years ago
- Code for paper "Real-time Neural Network Inference on Extremely Weak Devices: Agile Offloading with Explainable AI" (MobiCom'22)☆16Updated last year
- Reference code for https://arxiv.org/abs/1906.08879☆16Updated 5 years ago
- Self-Supervised Deep Learning based Surrogate Models for Fault-Tolerant Edge Computing☆19Updated last year
- Distributed Deep Graph Learning Framework for Dynamic Graphs☆11Updated 7 months ago
- PyTorch implementation of the paper: Multi-Agent Collaborative Inference via DNN Decoupling: Intermediate Feature Compression and Edge Le…☆27Updated last year
- ☆32Updated last year
- ☆15Updated 2 years ago