Zhang, Xuechen

14 publications

ICLR 2026 Continuous Chain of Thought Enables Parallel Exploration and Reasoning Halil Alperen Gozeten, Muhammed Emrullah Ildiz, Xuechen Zhang, Hrayr Harutyunyan, Ankit Singh Rawat, Samet Oymak
ICLR 2026 SmartChunk Retrieval: Query-Aware Chunk Compression with Planning for Efficient Document RAG Xuechen Zhang, Koustava Goswami, Samet Oymak, Jiasi Chen, Nedim Lipka
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