activity
20222025
most citedUMIX: Improving Importance Weighting for Subpopulation Shift via Uncertainty-Aware Mixup

10 citations · 13 across the 10 of their papers we have counts for

collaborators

10 papers

cs.AI2025

Multi-Task Vehicle Routing Solver via Mixture of Specialized Experts under State-Decomposable MDP

Yuxin Pan, Zhiguang Cao, Chengyang Gu +4

Existing neural methods for multi-task vehicle routing problems (VRPs) typically learn unified solvers to handle multiple constraints simultaneously. However, they often underutili…

cs.LG2025

CrystalDiT: A Diffusion Transformer for Crystal Generation

Xiaohan Yi, Guikun Xu, Xi Xiao +4

We present CrystalDiT, a diffusion transformer for crystal structure generation that achieves state-of-the-art performance by challenging the trend of architectural complexity. Ins…

cs.LG2025

Injecting Imbalance Sensitivity for Multi-Task Learning

Zhipeng Zhou, Liu Liu, Peilin Zhao +1

Multi-task learning (MTL) has emerged as a promising approach for deploying deep learning models in real-life applications. Recent studies have proposed optimization-based learning…

cs.LG2025

HDT: Hierarchical Discrete Transformer for Multivariate Time Series Forecasting

Shibo Feng, Peilin Zhao, Liu Liu +2

Generative models have gained significant attention in multivariate time series forecasting (MTS), particularly due to their ability to generate high-fidelity samples. Forecasting…

cs.CL2025★ 1 cited

Principled Data Selection for Alignment: The Hidden Risks of Difficult Examples

Chengqian Gao, Haonan Li, Liu Liu +3

The alignment of large language models (LLMs) often assumes that using more clean data yields better outcomes, overlooking the match between model capacity and example difficulty.…

cs.CL2024

Step-On-Feet Tuning: Scaling Self-Alignment of LLMs via Bootstrapping

Haoyu Wang, Guozheng Ma, Ziqiao Meng +9

Self-alignment is an effective way to reduce the cost of human annotation while ensuring promising model capability. However, most current methods complete the data collection and…

cs.LG2023

DFWLayer: Differentiable Frank-Wolfe Optimization Layer

Zixuan Liu, Liu Liu, Xueqian Wang +1

Differentiable optimization has received a significant amount of attention due to its foundational role in the domain of machine learning based on neural networks. This paper propo…

cs.LG2023

Reweighted Mixup for Subpopulation Shift

Zongbo Han, Zhipeng Liang, Fan Yang +8

Subpopulation shift exists widely in many real-world applications, which refers to the training and test distributions that contain the same subpopulation groups but with different…

cs.LG2023★ 2 cited

Deploying Offline Reinforcement Learning with Human Feedback

Ziniu Li, Ke Xu, Liu Liu +3

Reinforcement learning (RL) has shown promise for decision-making tasks in real-world applications. One practical framework involves training parameterized policy models from an of…

cs.LG2022★ 10 cited

UMIX: Improving Importance Weighting for Subpopulation Shift via Uncertainty-Aware Mixup

Zongbo Han, Zhipeng Liang, Fan Yang +7

Subpopulation shift widely exists in many real-world machine learning applications, referring to the training and test distributions containing the same subpopulation groups but va…