atlas

Research map / Probabilistic and scientific modeling / Dynamical systems

Emergence of Hidden Capabilities: Exploring Learning Dynamics in Concept Space

NeurIPS 2024 · Probabilistic and scientific modeling › Dynamical systems · approach: Latent dynamics models · 0 citations (OpenAlex)

Read the paperCheck your idea against 108,967 papers

AI-extracted summary. Written by AI from the paper's full text. Numbers and claims can be wrong or missing context, so check the paper before you cite or build on it.

Proposes a "concept space" framework and synthetic 2D-object toy datasets to study how diffusion models learn concepts, showing learning speed follows concept signal and that capabilities emerge suddenly before naive prompting can elicit them.

Problem

Understanding what determines which concepts a generative model learns, in what order, and when it acquires the ability to manipulate them, especially when capabilities are hidden from naive input prompting.

Approach

Analyze learning dynamics as trajectories in a concept space whose axes are independent concepts; show learning speed is set by concept signal (sensitivity of the data-generating process to each concept), and that sudden turns in trajectories coincide with hidden capabilities revealed by latent interventions or overprompting but not naive prompting.

Key results

Measured effects reported in the paper

What was changedWhereMeasuredEffect
Increase the data signal for a concept by widening its visual separation (ablation)Variational diffusion model on synthetic OOD classes · Synthetic color and size conceptslearning speed: 80% accuracyHelped
Make concept signals highly imbalanced (ablation)Variational diffusion model on synthetic OOD classes · Synthetic OOD class 11OOD generalization speedHurt
Increase the percentage of masked color prompts in training (ablation)Variational diffusion model · Synthetic blue triangle OOD classaccuracy: accuracy above 0.8Hurt
Use linear latent intervention instead of naive input promptingVariational diffusion model · Synthetic OOD class 11accuracy: around as early as 6,000 gradient stepsHelped
Use overprompting instead of naive input promptingVariational diffusion model · Synthetic OOD class 11accuracy: around as early as 6,000 gradient stepsHelped
Use overprompting on a model trained with 50% masked promptsVariational diffusion model · Synthetic OOD class 11accuracy: 100% accuracy after approximately 1,000 gradient stepsHelped
Use classifier-free guidance after the capability transition (ablation)Conditional diffusion model · Synthetic OOD generalizationOOD generalizationHelped
Train with underspecified prompts by masking color tokens (ablation)Variational diffusion model · Synthetic blue triangle OOD classOOD generalizationHurt
Patch the final-checkpoint conditioning embedding module into an intermediate checkpoint (ablation)Variational diffusion model · Synthetic OOD class 11accuracyMixed
Use latent interventions rather than naive promptingCelebA diffusion model · Female and With Hat OOD classOOD generalization: near $5\times10^5$ gradient stepsHelped
Increase a single concept's signal in the three-concept setting (ablation)Variational diffusion model · Synthetic three-concept OOD classesaccuracy on that conceptHelped
Vary batch size, channel count, learning rate, diffusion steps, weight decay, initialization scale, or optimizer (ablation)Variational diffusion model · Synthetic experimentsqualitative resultsNo clear effect

Surprising findings

Limitations

Experimental setup

Datasets: synthetic 2D shapes DGP (shape, color, size, background color) CelebA

Benchmarks: synthetic concept-space OOD class accuracy CelebA compositional generalization

Metrics: accuracy of concept classifier probes gradient steps to 80% accuracy 50% capability point

Baselines: naive input prompting classifier-free guidance Stable Diffusion v1.4

Builds on: variational diffusion models classifier-free guidance disentangled representation learning grokking

Closest papers by meaning

Writing a paper in this area? scime draft is a free LaTeX editor with AI included, real-time co-authors and a check of your references and venue rules before you submit. It also imports OpenAI Prism projects.

Open scime draft

Get the next scime Dispatch and a monthly field update

Free. We send a confirmation link first; every email has a one-click unsubscribe. Privacy