← Search

Jingchu Gai

6 accepted papers

2026

Demystifying Entropy Control in LLM RL Training: Theoretical Analysis and Dynamic Scheduling

ICML 2026spotlight

This paper investigates a pivotal yet debated component of reinforcement learning (RL) for training large language models (LLMs): controlling entropy (increasing or decreasing it) during RL fine-tuning. The existing literature presents a dichotomy: some studies posit that increasing entropy facilita…

Cited by 0SourceScholar
2026

Differential Smoothing Mitigates Sharpening and Improves LLM Reasoning

ICML 2026poster

It is widely recognized that reinforcement learning (RL) fine-tuning of large language models often leads to \textit{diversity collapse}, where outputs lack variety. Prior work has proposed a range of heuristics to counteract this effect, but these methods are ad hoc: they frequently trade off corre…

Cited by 0SourceScholar
2026

Towards Solving the Gilbert-Pollak Conjecture via Large Language Models

ICML 2026poster

The Gilbert-Pollak Conjecture, also known as the Steiner Ratio Conjecture, states that for any finite point set in the Euclidean plane, the Steiner minimum tree has length at least $\sqrt{3}/2 \approx 0.866$ times that of the Euclidean minimum spanning tree (the Steiner ratio). A sequence of improve…

Cited by 0SourceScholar
2025

Breaking the Curse of Multiagency in Robust Multi-Agent Reinforcement Learning

ICML 2025poster

Standard multi-agent reinforcement learning (MARL) algorithms are vulnerable to sim-to-real gaps. To address this, distributionally robust Markov games (RMGs) have been proposed to enhance robustness in MARL by optimizing the worst-case performance when game dynamics shift within a prescribed uncert…

Cited by 5SourcePDFScholar
2025

Homomorphism Expressivity of Spectral Invariant Graph Neural Networks

ICLR 2025oral

Graph spectra are an important class of structural features on graphs that have shown promising results in enhancing Graph Neural Networks (GNNs). Despite their widespread practical use, the theoretical understanding of the power of spectral invariants --- particularly their contribution to GNNs ---…

Cited by 0SourcePDFScholar
2024

Beyond Weisfeiler-Lehman: A Quantitative Framework for GNN Expressiveness

ICLR 2024oral

Designing expressive Graph Neural Networks (GNNs) is a fundamental topic in the graph learning community. So far, GNN expressiveness has been primarily assessed via the Weisfeiler-Lehman (WL) hierarchy. However, such an expressivity measure has notable limitations: it is inherently coarse, qualitati…