← Search

Yihe Deng

12 accepted papers

2026

Supervised Reinforcement Learning: From Expert Trajectories to Step-wise Reasoning

ICLR 2026poster

Large Language Models (LLMs) often struggle with challenging, multi-step reasoning problems due to a fundamental learning gap -- Reinforcement Learning with Verifiable Rewards (RLVR) suffers from sparse rewards when correct solutions are rarely sampled, while Supervised Fine-Tuning (SFT) tends to ov…

Cited by 0SourceScholar
2025

Mitigating Object Hallucination in Large Vision-Language Models via Image-Grounded Guidance

ICML 2025spotlight

The advancement of Large Vision-Language Models (LVLMs) has increasingly highlighted the critical issue of their tendency to hallucinate non-existing objects in the images. To address this issue, previous works focused on using specially curated datasets or powerful LLMs to rectify the outputs of LV…

2025

On the Power of Multitask Representation Learning with Gradient Descent

AISTATS 2025poster

Representation learning, particularly multi-task representation learning, has gained widespread popularity in various deep learning applications, ranging from computer vision to natural language processing, due to its remarkable generalization performance. Despite its growing use, our understanding…

Cited by 0SourceScholar
2025

OpenVLThinker: Complex Vision-Language Reasoning via Iterative SFT-RL Cycles

NeurIPS 2025poster

We introduce *OpenVLThinker*, one of the first open-source large vision–language models (LVLMs) to exhibit sophisticated chain-of-thought reasoning, achieving notable performance gains on challenging visual reasoning tasks. While text-based reasoning models (e.g., Deepseek R1) show promising results…

Cited by 0SourceScholar
2024

Enhancing Large Vision Language Models with Self-Training on Image Comprehension

NeurIPS 2024poster

Large vision language models (LVLMs) integrate large language models (LLMs) with pre-trained vision encoders, thereby activating the perception capability of the model to understand image inputs for different queries and conduct subsequent reasoning. Improving this capability requires high-quality v…

2024

GraphVis: Boosting LLMs with Visual Knowledge Graph Integration

NeurIPS 2024poster

The rapid evolution of large language models (LLMs) has expanded their capabilities across various data modalities, extending from well-established image data to increasingly popular graph data. Given the limitation of LLMs in hallucinations and inaccuracies in recalling factual knowledge, Knowledge…

Cited by 3SourcePDFScholar
2024

Risk Bounds of Accelerated SGD for Overparameterized Linear Regression

ICLR 2024poster

Accelerated stochastic gradient descent (ASGD) is a workhorse in deep learning and often achieves better generalization performance than SGD. However, existing optimization theory can only explain the faster convergence of ASGD, but cannot explain its better generalization. In this paper, we study t…

Cited by 8SourcePDFScholar
2024

Self-Play Fine-Tuning Converts Weak Language Models to Strong Language Models

ICML 2024poster

Harnessing the power of human-annotated data through Supervised Fine-Tuning (SFT) is pivotal for advancing Large Language Models (LLMs). In this paper, we delve into the prospect of growing a strong LLM out of a weak one without the need for acquiring additional human-annotated data. We propose a ne…

2024

Understanding Transferable Representation Learning and Zero-shot Transfer in CLIP

ICLR 2024poster

Multi-modal learning has become increasingly popular due to its ability to leverage information from different data sources (e.g., text and images) to improve the model performance. Recently, CLIP has emerged as an effective approach that employs vision-language contrastive pretraining to learn join…

Cited by 15SourcePDFScholar
2023

Robust Learning with Progressive Data Expansion Against Spurious Correlation

NeurIPS 2023poster

While deep learning models have shown remarkable performance in various tasks, they are susceptible to learning non-generalizable _spurious features_ rather than the core features that are genuinely correlated to the true label. In this paper, beyond existing analyses of linear models, we theoretica…

2022

Towards Understanding the Mixture-of-Experts Layer in Deep Learning

NeurIPS 2022accept

The Mixture-of-Experts (MoE) layer, a sparsely-activated model controlled by a router, has achieved great success in deep learning. However, the understanding of such architecture remains elusive. In this paper, we formally study how the MoE layer improves the performance of neural network learning…

Cited by 87SourcePDFScholar
2021

Adversarial Training with Fast Gradient Projection Method against Synonym Substitution Based Text Attacks

AAAI 2021technical

Adversarial training is the most empirically successful approach in improving the robustness of deep neural networks for image classification. For text classification, however, existing synonym substitution based adversarial attacks are effective but not very efficient to be incorporated into practi…