| 2026 | AAAI | HitKV: Activation Frequency Knows Which Tokens Are Important. | Sanle Zhao, Yujuan Tan, Jing Yu, Zhuoxin Bai, Yue Niu, Jiayi Guo, Zongjie Wang, Ao Ren |
| 2025 | EMNLP | MobiZO: Enabling Efficient LLM Fine-Tuning at the Edge via Inference Engines. | Lei Gao, Amir Ziashahabi, Yue Niu, Salman Avestimehr, Murali Annavaram |
| 2025 | VTC | LNN-Based Low-Complexity Secure Beamforming for RIS Assisted Wireless Communication Systems. | Qilong Lou, Jie Xu, Yue Niu, Yinghui Zhang |
| 2024 | CVPR | All Rivers Run to the Sea: Private Learning with Asymmetric Flows. | Yue Niu, Ramy E. Ali, Saurav Prakash, Salman Avestimehr |
| 2024 | IJCNN | RevPose: A Multi-Stage Reversible Networks for Human Pose Estimation. | Annan Wang, Yue Niu, Xuewu Wang, Shengxi Wu |
| 2024 | NAACL | Ethos: Rectifying Language Models in Orthogonal Parameter Space. | Lei Gao, Yue Niu, Tingting Tang, Salman Avestimehr, Murali Annavaram |
| 2023 | ICVS | Performance and Failure Cause Estimation for Machine Learning Systems in the Wild. | Xiruo Liu, Furqan Khan, Yue Niu, Pradeep Natarajan, Rinat Khaziev, Salman Avestimehr, Prateek Singhal |
| 2023 | LICS | A Metalanguage for Cost-Aware Denotational Semantics. | Yue Niu, Robert Harper |
| 2020 | FPGA | Reuse Kernels or Activations?: A Flexible Dataflow for Low-latency Spectral CNN Acceleration. | Yue Niu, Rajgopal Kannan, Ajitesh Srivastava, Viktor K. Prasanna |
| 2019 | HiPC | SPEC2: SPECtral SParsE CNN Accelerator on FPGAs. | Yue Niu, Hanqing Zeng, Ajitesh Srivastava, Kartik Lakhotia, Rajgopal Kannan, Yanzhi Wang, Viktor K. Prasanna |
| 2018 | LPAR | Automatic Space Bound Analysis for Functional Programs with Garbage Collection. | Yue Niu, Jan Hoffmann |