Zhang, Xuechen

12 publications

NeurIPS 2025 BREAD: Branched Rollouts from Expert Anchors Bridge SFT & RL for Reasoning Xuechen Zhang, Zijian Huang, Yingcong Li, Chenshun Ni, Jiasi Chen, Samet Oymak

AAAI 2025 On the Power of Convolution-Augmented Transformer Mingchen Li, Xuechen Zhang, Yixiao Huang, Samet Oymak

ICML 2025 Test-Time Training Provably Improves Transformers as In-Context Learners Halil Alperen Gozeten, Muhammed Emrullah Ildiz, Xuechen Zhang, Mahdi Soltanolkotabi, Marco Mondelli, Samet Oymak

AAAI 2024 A Score-Based Deterministic Diffusion Algorithm with Smooth Scores for General Distributions Karthik Elamvazhuthi, Xuechen Zhang, Matthew Jacobs, Samet Oymak, Fabio Pasqualetti

AAAI 2024 Class-Attribute Priors: Adapting Optimization to Heterogeneity and Fairness Objective Xuechen Zhang, Mingchen Li, Jiasi Chen, Christos Thrampoulidis, Samet Oymak

NeurIPS 2024 Efficient Contextual LLM Cascades Through Budget-Constrained Policy Learning Xuechen Zhang, Zijian Huang, Ege Onur Taga, Carlee Joe-Wong, Samet Oymak, Jiasi Chen

ICMLW 2024 On the Power of Convolution Augmented Transformer Mingchen Li, Xuechen Zhang, Yixiao Huang, Samet Oymak

NeurIPS 2024 Selective Attention: Enhancing Transformer Through Principled Context Control Xuechen Zhang, Xiangyu Chang, Mingchen Li, Amit Roy-Chowdhury, Jiasi Chen, Samet Oymak

NeurIPSW 2023 Augmenting Federated Learning with Pretrained Transformers Xuechen Zhang, Mingchen Li, Xiangyu Chang, Jiasi Chen, Amit Roy-Chowdhury, Ananda Suresh, Samet Oymak

L4DC 2023 Learning on Manifolds: Universal Approximations Properties Using Geometric Controllability Conditions for Neural ODEs Karthik Elamvazhuthi, Xuechen Zhang, Samet Oymak, Fabio Pasqualetti

NeurIPS 2023 Max-Margin Token Selection in Attention Mechanism Davoud Ataee Tarzanagh, Yingcong Li, Xuechen Zhang, Samet Oymak

NeurIPS 2021 AutoBalance: Optimized Loss Functions for Imbalanced Data Mingchen Li, Xuechen Zhang, Christos Thrampoulidis, Jiasi Chen, Samet Oymak