Preprints

Journal Publications

Conference Proceedings

• Understanding the Failure of Batch Normalization for Transformers in NLP
  Wang, Jiaxi and Ji, Wu and Lei, Huang
  2022 Conference on Neural Information Processing Systems (NeurIPS 2022) (Spotlight)

Interdisciplinary Publications

• Towards Predicting Equilibrium Distributions for Molecular Systems with Deep Learning
  Shuxin Zheng**, Jiyan He**, Chang Liu**, Yu Shi**, Ziheng Lu**, Weitao Feng, Fusong Ju, Jiaxi Wang , Jianwei Zhu, Yaosen Min, He Zhang, Shidi Tang, Hongxia Hao, Peiran Jin, Chi Chen, Frank Noé, Haiguang Liu, Tie-Yan Liu
  preprint

• Severity Evaluation of Obstructive Sleep Apnea Based on Speech Features
  Yiming Ding** and Jiaxi Wang** and Jiandong Gao and Qiang Fang and Yanru Li and Wen Xu and Ji Wu and Demin Han
  Sleep and Breathing(IF=2.816) 2021

• Bringing in the Outliers: A Sparse Subspace Clustering Approach to Learn a Dictionary of Mouse Ultrasonic Vocalizations
  Wang, Jiaxi and Mundnich, Karel and Knoll, Allison and Levitt, Pat and Narayanan,Shrikanth
  2020 IEEE International Conference on Acoustics,Speech, and Signal Processing (ICASSP 2020)