5 citations · 5 across the 9 of their papers we have counts for
5 papers · 1 filter
MLToolBench: Learning Tool-Augmented Agents for Machine Learning Development
Xin Yu, Lizhu Zhang, Jiamu Bai +7
Machine learning engineering (MLE) agents have made substantial progress, but learning through ML experimentation remains costly in time and computation. Synthetic environments red…
Difficulty-Aware Semantic-ID Optimization for Generative Recommendation
Xin Yu, Stephen Li, Sina Aghaei +6
Semantic-ID-based generative recommendation casts retrieval and ranking as autoregressive generation over hierarchical item identifiers. A common recipe is SFT followed by GRPO, ye…
Towards General Preference Alignment: Diffusion Models at Nash Equilibrium
Jiaming Hu, Jiamu Bai, Haoyu Wang +2
Reinforcement learning from human feedback (RLHF) has been popular for aligning text-to-image (T2I) diffusion models with human preferences. As a mainstream branch of RLHF, Direct…
Accurate and Scalable Matrix Mechanisms via Divide and Conquer
Guanlin He, Yingtai Xiao, Jiamu Bai +4
Matrix mechanisms are often used to provide unbiased differentially private query answers when publishing statistics or creating synthetic data. Recent work has developed matrix me…
Composition for Pufferfish Privacy
Jiamu Bai, Guanlin He, Xin Gu +2
When creating public data products out of confidential datasets, inferential/posterior-based privacy definitions, such as Pufferfish, provide compelling privacy semantics for data…