Selected Publications

Asterisk marks (*) indicate authors having contributed (almost) equally.

• Conformal Crystal Graph Transformer with Robust Encoding of Periodic Invariance
  Yingheng Wang, Shufeng Kong, John Gregoire, Carla Gomes
  AAAI 2024
  [Paper] [Code]

• ModuLoRA: Finetuning 2-Bit LLMs on Consumer GPUs by Integrating with Modular Quantizers
  Junjie Yin, Jiahao Dong, Yingheng Wang, Christopher De Sa, Volodymyr Kuleshov
  Transactions on Machine Learning Research (TMLR)
  [Paper] [Code]
  Featured Certification (presented at ICLR 2024)

• M$^2$Hub: Unlocking the Potential of Machine Learning for Materials Discovery
  Yuanqi Du*, Yingheng Wang*, Yining Huang, Jianan Canal Li, Yanqiao Zhu, Tian Xie, Chenru Duan, John Gregoire, Carla Gomes
  NeurIPS 2023 Dataset and Benchmark
  [Paper] [Code]

• InfoDiffusion: Representation Learning Using Information Maximizing Diffusion Models
  Yingheng Wang, Yair Schiff, Aaron Gokaslan, Weishen Pan, Fei Wang, Christopher De Sa, Volodymyr Kuleshov
  ICML 2023
  [Paper] [Code]

• Graph-Enhanced Emotion Neural Decoding
  Zhongyu Huang, Changde Du, Yingheng Wang, Kaicheng Fu, Huiguang He
  IEEE Transactions on Medical Imaging
  [Paper]

• Time Series Contrastive Learning with Information-Aware Augmentations
  Dongsheng Luo, Wei Cheng, Yingheng Wang, Dongkuan Xu, Jingchao Ni, Wenchao Yu, Xuchao Zhang, Yanchi Liu, Yuncong Chen, Haifeng Chen, Xiang Zhang
  AAAI 2023
  [Paper] [Code]

• Going Deeper into Permutation-sensitive Graph Neural Networks
  Zhongyu Huang*, Yingheng Wang*, Chaozhuo Li, Huiguang He
  ICML 2022
  [Paper] [Code]

• Graph Emotion Decoding from Visually Evoked Neural Responses
  Zhongyu Huang, Changde Du, Yingheng Wang, Huiguang He
  MICCAI 2022
  [Paper]

• Multi-view Graph Contrastive Representation Learning for Drug-drug Interaction Prediction
  Yingheng Wang, Yaosen Min, Xin Chen, Ji Wu
  WWW 2021
  [Paper] [Code]
  A previous version won Best Paper Award at DLG-AAAI 2021

• Molecular Graph Contrastive Learning with Parameterized Explainable Augmentations
  Yingheng Wang*, Yaosen Min*, Erzhuo Shao, Ji Wu
  BIBM 2021
  [Paper]

• MolCloze: A Unified Cloze-style Self-supervised Molecular Structure Learning Model for Chemical Property Prediction
  Yingheng Wang*, Xin Chen*, Yaosen Min, Ji Wu
  BIBM 2021
  [Paper]

• One-shot Transfer Learning for Population Mapping
  Erzhuo Shao*, Jie Feng*, Yingheng Wang, Tong Xia, Yong Li
  CIKM 2021
  [Paper] [Code]