Skip to content

Author

Jiamu Bai

We have 4 of 13 papers

We haven’t gathered this author’s papers yet. Follow them and we’ll fetch their work.

Not the right person? Other researchers publish under this name.

#artificial intelligence Preprint Oct 2026

Personalize at Test Time: Learning User Preferences for Image Generation

Diffusion models can generate high-quality images, yet aligning their outputs with individual user preferences remains challenging. A key bottleneck is accurately modeling diverse user preferences from limited feedback. Existing approaches often rely on labor-intensive manual preference annotations or vision-language m...

Jiamu Bai, Jia-Ming Hu, Yan-Hong Wu et al. · 0 citations
#artificial intelligence Preprint Oct 2026

Verify Less, Evolve More: Training Idea-Level Critics for Verification-Efficient ML Evolving Agents

As large language models become more powerful, self-evolving agents are able to tackle challenging tasks including AI for machine learning (AI4ML). In AI4ML, while empirical verification is available, it often requires computationally costly model training and evaluation, limiting the speed and scale of agent evolution...

Jiamu Bai, Li-Zhu Zhang, Xin Yu et al. · 0 citations
#artificial intelligence Preprint Sep 2026

MLToolBench: Learning Tool-Augmented Agents for Machine Learning Development

This work introduces ToolMLBench, a suite of executable tools for data inspection, code verification, and experiment diagnosis, together with an SFT and RL pipeline for learning their use, and measures how privileged context changes the likelihood of a sampled tool action and uses this difference as a turn-level reward...

Xin Yu, Li-Zhu Zhang, Jiamu Bai et al. · 0 citations
Preprint Aug 2026

Difficulty-Aware Semantic-ID Optimization for Generative Recommendation

Difficulty-Aware Semantic-ID Optimization (DASO), a tree-aware post-training method that addresses failure mode as an online rollout-allocation problem and improves over MiniOneRec-style GRPO on 11 of 12 metrics and achieves the best result on 9 of 12 metrics.

Xin Yu, Stephen Li, S. Aghaei et al. · 1 citation

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.