Skip to content

@MIRALab-USTC MIRA Lab

Change the repository type filter

All

    Repositories list

    44 repositories

    • AI4EDA_TNet

      Public
      Python
      •1•3•0•0•Updated Jan 10, 2025Jan 10, 2025

    • AI4Sci-MiCaM

      Public
      This is the code of paper "De Novo Molecular Generation via Connection-aware Motif Mining". Zijie Geng, Shufang Xie, Yingce Xia, Lijun Wu, Tao Qin, Jie Wang, Yongdong Zhang, Feng Wu, Tie-Yan Liu. ICLR 2023.
      Python
      •10•54•4•0•Updated Jan 7, 2025Jan 7, 2025

    • RL-CRESP

      Public
      Python
      •1•6•1•0•Updated Dec 17, 2024Dec 17, 2024

    • ChiPBench

      Public
      ChiPBench:Benchmarking End-to-End Performance of AI-based Chip Placement Algorithms
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •1•25•1•0•Updated Nov 28, 2024Nov 28, 2024

    • AI4EDA-EfficientPlace

      Public
      This is the code for our paper "Reinforcement Learning within Tree Search for Fast Macro Placement".
      Python
      •
      MIT License
      •1•15•0•0•Updated Nov 13, 2024Nov 13, 2024

    • RL-TRACER

      Public
      Python
      •
      MIT License
      •0•6•0•0•Updated Oct 15, 2024Oct 15, 2024

    • L2O-G2MILP

      Public
      This is the code for G2MILP, a deep learning-based mixed-integer linear programming (MILP) instance generator.
      Python
      •
      MIT License
      •5•29•0•0•Updated Oct 3, 2024Oct 3, 2024

    • L2O-GS4CO

      Public
      The code of paper *Towards General Algorithm Discovery for Combinatorial Optimization: Learning Symbolic Branching Policy from Bipartite Graph*.
      Python
      •0•0•0•0•Updated Aug 13, 2024Aug 13, 2024

    • L2O-Symb4CO

      Public
      The code of paper *Rethinking Branching on Exact Combinatorial Optimization Solver: The First Deep Symbolic Discovery Framework*.
      Python
      •0•4•0•0•Updated Apr 19, 2024Apr 19, 2024

    • scip

      Public
      The scip codes used for machine learning purposes. Downloaded from https://www.scipopt.org/index.php#download
      C
      •
      Other
      •0•0•0•0•Updated Apr 19, 2024Apr 19, 2024

    • PySCIPOpt

      Public
      The PySCIPOpt codes used for machine learning purposes. Forked from https://github.com/ds4dm/PySCIPOpt.git .
      Python
      •
      MIT License
      •257•0•0•0•Updated Apr 18, 2024Apr 18, 2024

    • UKB-MITH

      Public
      Python
      •0•0•0•0•Updated Nov 25, 2023Nov 25, 2023

    • RL-SPF

      Public
      a representation learning method that predicts the Fourier transform of state sequences to improve sample efficiency of RL algorithms.
      Python
      •1•18•0•0•Updated Oct 26, 2023Oct 26, 2023

    • LD

      Public
      Python
      •0•6•0•0•Updated Sep 27, 2023Sep 27, 2023

    • AI4LogicSynthesis-PruneX

      Public
      Python
      •1•9•0•0•Updated Aug 22, 2023Aug 22, 2023

    • RL-CBM

      Public
      The code of paper Robust Representation learning by Clustering with Bisimulation Metrics for Visual Reinforcement Learning with Distractions. Qiyuan Liu, Qi Zhou, Rui Yang, Jie Wang. AAAI 2023.
      Python
      •1•6•0•0•Updated May 23, 2023May 23, 2023

    • RL-RAEB

      Public
      This is the code for the paper "Efficient Exploration in Resource-Restricted Reinforcement Learning" (https://arxiv.org/abs/2212.06988)
      Python
      •1•3•0•0•Updated May 13, 2023May 13, 2023

    • L2O-HEM-Torch

      Public
      The code of paper Learning Cut Selection for Mixed-Integer Linear Programming via Hierarchical Sequence Model. Zhihai Wang, Xijun Li, Jie Wang*, Yufei Kuang, Mingxuan Yuan, Jia Zeng, Yongdong Zhang, Feng Wu. ICLR 2023.
      Python
      •
      MIT License
      •8•55•2•0•Updated May 12, 2023May 12, 2023

    • GNN-LMC

      Public
      The code of paper LMC: Fast Training of GNNs via Subgraph Sampling with Provable Convergence. Zhihao Shi, Xize Liang, Jie Wang. ICLR 2023.
      Python
      •3•43•1•0•Updated Feb 15, 2023Feb 15, 2023

    • GraphAKD

      Public
      The code of paper Compressing Deep Graph Neural Networks via Adversarial Knowledge Distillation. Huarui He, Jie Wang, Zhanqiu Zhang, Feng Wu. SIGKDD 2022.
      Python
      •
      MIT License
      •3•41•0•0•Updated Oct 18, 2022Oct 18, 2022

    • DD-RetroDCVAE

      Public
      A novel template-free retrosynthesizer that can generate diverse sets of reactants for a desired product via discrete conditional variational autoencoders.
      Python
      •
      MIT License
      •1•14•0•0•Updated Aug 7, 2022Aug 7, 2022

    • DCRN

      Public
      This is the code of paper Deep Cognitive Reasoning Network for Multi-hop Question Answering over Knowledge Graphs. Jianyu Cai, Zhanqiu Zhang, Feng Wu, Jie Wang. Findings of ACL 2021
      knowledge-graph
      Python
      •
      BSD 2-Clause "Simplified" License
      •2•2•1•0•Updated May 30, 2022May 30, 2022

    • RL-CMBAC

      Public
      The code of paper Sample-Efficient Reinforcement Learning via Conservative Model-Based Actor-Critic. Zhihai Wang, Jie Wang*, Qi Zhou, Bin Li, Houqiang Li. AAAI 2022.
      Python
      •0•20•0•0•Updated May 26, 2022May 26, 2022

    • KG-TACT

      Public
      The code of paper Topology-Aware Correlations Between Relations for Inductive Link Prediction in Knowledge Graphs. Jiajun Chen, Huarui He, Feng Wu, Jie Wang. AAAI 2021.
      knowledge-graphinductive-link-prediction
      Python
      •7•42•1•0•Updated May 11, 2022May 11, 2022

    • GCN4KGC

      Public
      The code of paper Rethinking Graph Convolutional Networks in Knowledge Graph Completion. Zhanqiu Zhang, Jie Wang, Jieping Ye, Feng Wu. WWW 2022.
      Python
      •11•61•1•0•Updated Mar 29, 2022Mar 29, 2022

    • RL-SCPO

      Public
      The code of paper *Learning Robust Policy against Disturbance in Transition Dynamics via State-Conservative Policy Optimization*.
      Python
      •4•18•0•0•Updated Mar 26, 2022Mar 26, 2022

    • QE-ConE

      Public
      The code of paper ConE: Cone Embeddings for Multi-Hop Reasoning over Knowledge Graphs. Zhanqiu Zhang, Jie Wang, Jiajun Chen, Shuiwang Ji, Feng Wu. NeurIPS 2021.
      Python
      •
      MIT License
      •8•45•0•0•Updated Feb 8, 2022Feb 8, 2022

    • KGE-DURA

      Public
      The code of paper Duality-Induced Regularizer for Tensor Factorization Based Knowledge Graph Completion. Zhanqiu Zhang, Jianyu Cai, Jie Wang. NeurIPS 2020.
      knowledge-graphknowledge-graph-embeddingtensor-factorizationknowledge-graph-completionkge-dura
      Python
      •9•50•1•0•Updated Dec 22, 2021Dec 22, 2021

    • KDDCup2021_WikiKG90M_GraphMIRAcles

      Public
      Python
      •
      MIT License
      •2•8•0•0•Updated Jul 26, 2021Jul 26, 2021

    • KGE-HAKE

      Public
      The code of paper Learning Hierarchy-Aware Knowledge Graph Embeddings for Link Prediction. Zhanqiu Zhang, Jianyu Cai, Yongdong Zhang, Jie Wang. AAAI 2020.
      knowledge-graphknowledge-graph-completionknowledge-graph-embeddings
      Python
      •
      MIT License
      •53•252•3•0•Updated Apr 12, 2021Apr 12, 2021