← Search

Jing Bai

15 accepted papers

2026

EEPO: Exploration-Enhanced Policy Optimization via Sample-Then-Forget

ICLR 2026poster

Balancing exploration and exploitation remains a central challenge in reinforcement learning with verifiable rewards (RLVR) for large language models (LLMs). Current RLVR methods often overemphasize exploitation, leading to entropy collapse, reduced exploratory capacity, and ultimately limited perfo…

Cited by 0SourcecodeScholar
2026

Learning Useful Supervision for Reinforcement Learning in Reasoning Models

ICML 2026poster

Supervised fine-tuning (SFT) and reinforcement learning with verifiable rewards (RLVR) are two widely used post-training paradigms for improving the reasoning ability of large language models (LLMs). Recent methods attempt to integrate SFT and RLVR in a single stage by reweighting or scheduling thei…

Cited by 0SourceScholar
2026

ReVeal: Self-Evolving Code Agents via Reliable Self-Verification

ICLR 2026poster

Reinforcement learning with verifiable rewards (RLVR) has advanced the reasoning capabilities of large language models. Howerer, existing methods rely solely on outcome rewards, without explicitly optimizing verification or leveraging reliable signals from realistic environments, leading to unreliab…

Cited by 0SourceScholar
2025

DreamBench++: A Human-Aligned Benchmark for Personalized Image Generation

ICLR 2025poster

Personalized image generation holds great promise in assisting humans in everyday work and life due to its impressive function in creatively generating personalized content. However, current evaluations either are automated but misalign with humans or require human evaluations that are time-consumin…

2025

FG3DFormer: Fine-Grained 3D Shape Classification Based on Vision Transformer

ICASSP 2025accepted

Fine-grained 3D shape classification (FGSC) remains challenging due to the difficulty of adaptively capturing global structure differences and subtle inter-class distinctions. This paper directly extends Vision Transformer (ViT) to FGSC, proposing a pure Transformer network FG3DFormer that fully lev…

Cited by 0SourceScholar
2025

Teaching LLMs to Plan, Not Just Solve: Plan Learning Boosts LLMs Generalization in Reasoning Tasks

EMNLP 2025

Reinforcement learning (RL) on self-generated data has emerged as a promising paradigm for improving reasoning in large language models (LLMs). However, RL relies on accurate reward signals, which are scarce in many domains, making it critical to train models that can generalize to unseen problems.

2025

Vulnerability-Aware Alignment: Mitigating Uneven Forgetting in Harmful Fine-Tuning

ICML 2025poster

Harmful fine-tuning (HFT), performed directly on open-source LLMs or through Fine-tuning-as-a-Service, breaks safety alignment and poses significant threats. Existing methods aim to mitigate HFT risks by learning robust representation on alignment data or making harmful data unlearnable, but they tr…

Cited by 0SourcePDFScholar
2024

AdapterGNN: Parameter-Efficient Fine-Tuning Improves Generalization in GNNs

AAAI 2024technical

Fine-tuning pre-trained models has recently yielded remarkable performance gains in graph neural networks (GNNs). In addition to pre-training techniques, inspired by the latest work in the natural language fields, more recent work has shifted towards applying effective fine-tuning approaches, such a…

2022

Creating Training Sets via Weak Indirect Supervision

ICLR 2022poster

Creating labeled training sets has become one of the major roadblocks in machine learning. To address this, recent Weak Supervision (WS) frameworks synthesize training labels from multiple potentially noisy supervision sources. However, existing frameworks are restricted to supervision sources that…

Cited by 19SourcePDFScholar
2022

Enhancing Self-Attention with Knowledge-Assisted Attention Maps

NAACL 2022long

Large-scale pre-trained language models have attracted extensive attentions in the research community and shown promising results on various tasks of natural language processing. However, the attention maps, which record the attention scores between tokens in self-attention mechanism, are sometimes…

Cited by 8SourcePDFScholar
2021

Evolving Attention with Residual Convolutions

ICML 2021spotlight

Transformer is a ubiquitous model for natural language processing and has attracted wide attentions in computer vision. The attention maps are indispensable for a transformer model to encode the dependencies among input tokens. However, they are learned independently in each layer and sometimes fail…

2020

LadaBERT: Lightweight Adaptation of BERT through Hybrid Model Compression

COLING 2020main

BERT is a cutting-edge language representation model pre-trained by a large corpus, which achieves superior performances on various natural language understanding tasks. However, a major blocking issue of applying BERT to online services is that it is memory-intensive and leads to unsatisfactory lat…

2020

Spectral Temporal Graph Neural Network for Multivariate Time-series Forecasting

NeurIPS 2020spotlight

Multivariate time-series forecasting plays a crucial role in many real-world applications. It is a challenging problem as one needs to consider both intra-series temporal correlations and inter-series correlations simultaneously. Recently, there have been multiple works trying to capture both correl…