Organizations: Department of Computer and Information Science, Fordham University, New York, NY · Department of Neurology, Mass General Brigham, Boston, MA
Language models (LMs) and classical machine learning methods offer complementary strengths for predictive modeling, yet their fundamentally different representations and training paradigms hinder effective integration: LMs rely on gradient-based optimization over textual data, whereas models such as Random Forests (RF) employ non-differentiable feature partitioning. This work introduces a reciprocal co-training framework that couples an LM with an RF classifier via reinforcement learning, creating an iterative feedback loop in which each model improves using signals from the other. Tabular data are reformulated into standardized textual representations for the LM, whose embeddings augment the RF feature space, while calibrated RF probability estimates provide feedback signals that guide reinforcement learning updates of the LM. Experiments across three medical datasets, evaluated with both a domain-adapted clinical encoder (ClinicalBERT) and a larger instruction-tuned language model (Qwen2-7B-Instruct), demonstrate consistent performance gains for both model components. Ablation analyses indicate that iterative refinement, hybrid reward design, and dimensionality control jointly contribute to these gains. SHAP analysis further confirms that LM-derived representations are among the most important inputs to the RF predictions. The proposed framework provides a general mechanism that allows incompatible model families to leverage each other's strengths through bidirectional adaptation.
Figures & tables
Fig. 2: Overview of the Reciprocal Co-Training (RCT) framework. Blue arrows indicate forward data flow, while orange arrows indicate reward feedback for PPO updates. The dashed loop denotes one RCT iteration.
Category
Model
ROC-AUC
PR-AUC
F1
Accuracy
Precision
Specificity
Recall
MS Dataset ( n=2192 )
Baselines
DT
0.635
0.443
0.558
0.529
0.419
0.360
0.837
RF
0.686
0.505
0.576
0.578
0.449
0.453
0.806
XGB
0.685
0.504
0.581
0.585
0.455
0.464
0.806
CBERT
0.672
0.525
0.595
0.589
0.472
0.461
0.805
Qwen2
0.660
0.536
0.563
0.532
0.433
0.370
0.805
TABLE I: Predictive Performance of the RCT Framework Compared with Standalone Baselines on Three Datasets
Fig. 3: ROC (left) and precision–recall (right) curves for relapse prediction on the MS dataset using CBERT. The proposed RCT model consistently dominates the baselines across most thresholds.
Fig. 4: Training dynamics of the RCT process on the MS-Relapse dataset using CBERT. (a) ROC-AUC of the RF and LM across outer iterations of alternating optimization. (b) Mean PPO reward during LM updates.
Dataset
Model
LM
Iter
RF
Iter
MS
CBERT
0.700
11
0.721
7
Qwen2
0.705
12
0.720
8
WDBC
CBERT
0.999
18
0.999
18
Qwen2
0.985
1
0.998
1
BRFSS
CBERT
0.826
5
0.829
5
Qwen2
0.813
6
0.823
1
TABLE II: Convergence Summary Across Datasets and Language Models
Component
Variant
ROC-AUC ( Δ )
LM-focused ablations
Training Scheme
Iterative (Full RCT)
0.700
One reciprocal iteration
0.694 (-0.006)
Optimization
PPO
0.694
REINFORCE
0.692 (-0.002)
Reward
Hybrid ( RRF+Racc )
0.606
TABLE III: Ablation study on the MS-Relapse dataset
Fig. 5: Top-20 RF features on the MS-Relapse dataset ranked by mean absolute SHAP value. LM-derived PCA components and original clinical variables are reported. Left: CBERT. Right: Qwen2.
Reward models (RMs) are central to aligning large language models (LLMs) with human values, yet noisy preference feedback can induce reward misgeneralization through spurious correlations and overfitting. We characterize the training dynamics of preference pairs and uncover a distinctive optimization signature of noise: persistently high losses, sharp temporal fluctuations, and irregular gradient updates. Our analyses further show that excluding such pairs improves downstream alignment. In practice, however, reliable preferences are not known a priori, and self-filtering based on a model's own training signals can amplify confirmation bias. To address this challenge, we introduce Collaborative Reward Modeling (CRM), an online framework for identifying reliable preferences. CRM couples cross-model peer review with curriculum learning: two RMs evaluate each other's sample selections, while curriculum keeps their learning progress synchronized. Across extensive experiments, CRM improves RM generalization by up to 9.94 points on RewardBench under 40% preference noise. CRM further generalizes to implicit-reward alignment, offering a robust and broadly applicable approach to learning from imperfect feedback.
Jiazheng Zhang, Wenqing Jing, Zizhuo Zhang +12
NLP Group, Fudan University · Hong Kong Baptist University · LLM Team, Meituan Inc
Large language models (LLMs) are increasingly used to automate AI research, including language-model (LM) development, offering a potential path toward recursive self-improvement. However, turning experimental feedback into a learned ability to propose effective improvements across models faces two challenges. First, training such an ability with real feedback is expensive: reinforcement learning (RL) requires evaluating tens of thousands of proposals, each potentially requiring a full model-training run. Even at ten GPU-hours per run, reward collection alone would consume hundreds of thousands of GPU-hours, motivating a cheaper surrogate. Second, surrogate feedback can become unreliable as policy optimization favors overestimated proposals when the proposal distribution shifts. To address these challenges, we propose LM2LM, which trains LLMs with RL to generate effective LM improvement proposals. To provide scalable feedback, we use a research world model (RWM), a frozen LLM that predicts proposal gains from real experimental records provided in context, as the reward model for RL. To maintain feedback reliability, we periodically execute current-policy proposals ranked highly by the RWM and add their measured outcomes to its context. Across eight autoregressive and diffusion LM development environments, real-execution evaluations on held-out tasks show that LM2LM enables Qwen3.8-27B to outperform GPT-6-sol, even when GPT-6-sol is given 300 real experimental records in context, while requiring 33% lower data-generation cost. LM2LM also generalizes to a model family unseen during training.
Minhua Lin, Zijun Wang, Zewen Liu +7
The Pennsylvania State University · UC Santa Cruz · Emory University +2
Policy loss estimation remains a fundamental and long-standing challenge in reinforcement learning (RL) for diffusion language models (DLMs). We introduce Reinforcement Learning from Denoising Feedback (RLDF), a novel training paradigm that leverages feedback obtained from rollout and training processes to facilitate accurate and efficient policy loss estimation. To balance the trade-off between computational efficiency and estimation effectiveness, RLDF optimizes the model toward the clipped clean state from intermediate noisy states, combined with weighted timestep sampling over denoising timesteps. Extensive experiments demonstrate that RLDF achieves consistent and substantial improvements in both performance and generalizability across two representative DLM architectures, LLaDA and Dream, on multiple reasoning benchmarks. Our work lays a principled foundation for scalable reinforcement learning in diffusion language models. We build Drift, a training framework for DLMs, available at https://github.com/ant-research/Drift.
Qi He, Huan Chen, Ya Guo +3
Fudan University · Ant Group · Hong Kong University of Science and Technology