8 papers
Generative Retrieval via Diffusion Transformer with Metric-Ordered Sequence Training and Hybrid-Policy Preference Optimization
Chenghao Liu, Yu Zhang, Zhongtao Jiang +7
Embedding-based retrieval ranks items by their similarity to a query in a shared vector space and usually aims to return the highest-scoring items. In many production settings this…
Stabilizing black-box algorithms through task-oriented randomization
Yali Wang, Zhaojun Wang
As black-box models become foundational to modern research, ensuring their stability is paramount for the realization of trustworthy artificial intelligence. The inherent diversity…
Shape-Adaptive Conditional Calibration for Conformal Prediction via Minimax Optimization
Yajie Bao, Chuchen Zhang, Zhaojun Wang +2
Achieving valid conditional coverage in conformal prediction is challenging due to the theoretical difficulty of satisfying pointwise constraints in finite samples. Building upon t…
Feedback-Enhanced Online Multiple Testing with Applications to Conformal Selection
Lin Lu, Yuyang Huo, Haojie Ren +2
This work studies online multiple testing with feedback, where decisions are made sequentially, and the true state of the hypothesis is revealed after decisions are made, either in…
Conditional Rank-Rank Regression via Deep Conditional Transformation Models
Xiaoyi Wang, Long Feng, Zhaojun Wang
Intergenerational mobility quantifies the transmission of socio-economic outcomes from parents to children. While rank-rank regression (RRR) is standard, adding covariates directly…
Aggregating Conformal Prediction Sets via α-Allocation
Congbin Xu, Yue Yu, Haojie Ren +2
Conformal prediction offers a distribution-free framework for constructing prediction sets with finite-sample coverage. Yet, efficiently leveraging multiple conformity scores to re…