Secrecy capacity and wireless energy harvesting in amplify-and-forward relay networks. - DiJia Su; S. A. Mousavifar; C. Leung (PACRIM, 2015)

ConQUR: Mitigating Delusional Bias in Deep Q-Learning. - Dijia Su; Jayden Ooi; Tyler Lu; Dale Schuurmans; Craig Boutilier (ICML, 2020)

MUSBO: Model-based Uncertainty Regularized and Sample Efficient Batch Optimization for Deployment Constrained Reinforcement Learning. - DiJia Su; Jason D. Lee; John M. Mulvey; H. Vincent Poor (CoRR, 2021)

PoBRL: Optimizing Multi-Document Summarization by Blending Reinforcement Learning Policies. - DiJia Andy Su; Difei Su; John M. Mulvey; H. Vincent Poor (CoRR, 2021)

Competitive Multi-Agent Reinforcement Learning with Self-Supervised Representation. - DiJia Su; Jason D. Lee; John M. Mulvey; H. Vincent Poor (ICASSP, 2022)

Narrowing the coordinate-frame gap in behavior prediction models: Distillation for efficient and accurate scene-centric motion forecasting. - DiJia Andy Su; Bertrand Douillard; Rami Al-Rfou; Cheol Park; Benjamin Sapp (ICRA, 2022)

Narrowing the Coordinate-frame Gap in Behavior Prediction Models: Distillation for Efficient and Accurate Scene-centric Motion Forecasting. - DiJia Su; Bertrand Douillard; Rami Al-Rfou; Cheolho Park; Benjamin Sapp (CoRR, 2022)

A Deep Reinforcement Learning Approach for Finding Non-Exploitable Strategies in Two-Player Atari Games. - Zihan Ding; Dijia Su; Qinghua Liu; Chi Jin 001 (CoRR, 2022)

Topics in Machine Learning - DiJia Su (2022)

Optimizing Multidocument Summarization by Blending Reinforcement Learning Policies. - DiJia Su; Difei Su; John M. Mulvey; H. Vincent Poor (IEEE Trans. Artif. Intell., 2023)

Training Large Language Models to Reason in a Continuous Latent Space. - Shibo Hao; Sainbayar Sukhbaatar; DiJia Su; Xian Li 003; Zhiting Hu; Jason Weston; Yuandong Tian (CoRR, 2024)

Dualformer: Controllable Fast and Slow Thinking by Learning with Randomized Reasoning Traces. - DiJia Su; Sainbayar Sukhbaatar; Michael Rabbat; Yuandong Tian; Qinqing Zheng (ICLR, 2025)

Token Assorted: Mixing Latent and Text Tokens for Improved Language Model Reasoning. - DiJia Su; Hanlin Zhu; Yingchen Xu; Jiantao Jiao; Yuandong Tian; Qinqing Zheng (CoRR, 2025)

GaLore 2: Large-Scale LLM Pre-Training by Gradient Low-Rank Projection. - DiJia Su; Andrew Gu; Jane Xu; Yuandong Tian; Jiawei Zhao (CoRR, 2025)