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
Showing cs.LGShow all

7 papers · 1 filter

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.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…