← Search

Dingli Yu

12 accepted papers

2025

Generalizing from SIMPLE to HARD Visual Reasoning: Can We Mitigate Modality Imbalance in VLMs?

ICML 2025poster

Vision Language Models (VLMs) are impressive at visual question answering and image captioning. But they underperform on multi-step visual reasoning---even compared to LLMs on the same tasks presented in text form---giving rise to perceptions of *modality imbalance* or *brittleness*. Towards a syste…

2025

Weak-to-Strong Generalization Even in Random Feature Networks, Provably

ICML 2025poster

Weak-to-Strong Generalization (Burns et al.,2024) is the phenomenon whereby a strong student, say GPT-4, learns a task from a weak teacher, say GPT-2, and ends up significantly outperforming the teacher. We show that this phenomenon does not require a complex and pretrained learner like GPT-4, can a…

Cited by 0SourcePDFScholar
2024

Can Models Learn Skill Composition from Examples?

NeurIPS 2024poster

As large language models (LLMs) become increasingly advanced, their ability to exhibit compositional generalization---the capacity to combine learned skills in novel ways not encountered during training---has garnered significant attention. This type of generalization, particularly in scenarios beyo…

Cited by 3SourcePDFScholar
2024

ConceptMix: A Compositional Image Generation Benchmark with Controllable Difficulty

NeurIPS 2024poster

Compositionality is a critical capability in Text-to-Image (T2I) models, as it reflects their ability to understand and combine multiple concepts from text descriptions. Existing evaluations of compositional capability rely heavily on human-designed text prompts or fixed templates, limiting their di…

Cited by 11SourcePDFScholar
2024

Keeping LLMs Aligned After Fine-tuning: The Crucial Role of Prompt Templates

NeurIPS 2024poster

Public LLMs such as the Llama 2-Chat underwent alignment training and were considered safe. Recently Qi et al. (2024) reported that even benign fine-tuning on seemingly safe datasets can give rise to unsafe behaviors in the models. The current paper is about methods and best practices to mitigate su…

2024

SKILL-MIX: a Flexible and Expandable Family of Evaluations for AI Models

ICLR 2024poster

With LLMs shifting their role from statistical modeling of language to serving as general-purpose AI agents, how should LLM evaluations change? Arguably, a key ability of an AI agent is to flexibly combine, as needed, the basic skills it has learned. The capability to combine skills plays an importa…

2024

Tensor Programs VI: Feature Learning in Infinite Depth Neural Networks

ICLR 2024poster

Empirical studies have consistently demonstrated that increasing the size of neural networks often yields superior performance in practical applications. However, there is a lack of consensus regarding the appropriate scaling strategy, particularly when it comes to increasing the depth of neural net…

Cited by 45SourcePDFScholar
2023

A Kernel-Based View of Language Model Fine-Tuning

ICML 2023poster

It has become standard to solve NLP tasks by fine-tuning pre-trained language models (LMs), especially in low-data settings. There is minimal theoretical understanding of empirical success, e.g., why fine-tuning a model with $10^8$ or more parameters on a couple dozen training points does not result…

2022

Fast Mixing of Stochastic Gradient Descent with Normalization and Weight Decay

NeurIPS 2022accept

We prove the Fast Equilibrium Conjecture proposed by Li et al., (2020), i.e., stochastic gradient descent (SGD) on a scale-invariant loss (e.g., using networks with various normalization schemes) with learning rate $\eta$ and weight decay factor $\lambda$ mixes in function space in $\mathcal{\tilde{…

Cited by 23SourcePDFScholar
2022

New Definitions and Evaluations for Saliency Methods: Staying Intrinsic, Complete and Sound

NeurIPS 2022accept

Saliency methods compute heat maps that highlight portions of an input that were most important for the label assigned to it by a deep net. Evaluations of saliency methods convert this heat map into a new masked input by retaining the $k$ highest-ranked pixels of the original input and replacing the…

2020

Harnessing the Power of Infinitely Wide Deep Nets on Small-data Tasks

ICLR 2020spotlight

Recent research shows that the following two models are equivalent: (a) infinitely wide neural networks (NNs) trained under l2 loss by gradient descent with infinitesimally small learning rate (b) kernel regression with respect to so-called Neural Tangent Kernels (NTKs) (Jacot et al., 2018). An effi…

Cited by 202SourcecodeScholar
2020

Simple and Effective Regularization Methods for Training on Noisily Labeled Data with Generalization Guarantee

ICLR 2020poster

Over-parameterized deep neural networks trained by simple first-order methods are known to be able to fit any labeling of data. Such over-fitting ability hinders generalization when mislabeled training examples are present. On the other hand, simple regularization methods like early-stopping can oft…

Cited by 91SourceScholar