Portrait of Toni J.B. Liu

Toni J.B. Liu

Research Highlights

A curated selection, not a full publication list.

  1. Direction of Ignorance
  2. Jacobian Scopes for Causal Attribution
  3. LLMs think like a kernel machine
  4. Memory–abstraction trade-off in diffusion models
  5. In-context neural scaling law
  6. Embedding and attention in non-Euclidean spaces

Direction of Ignorance

What does an LLM predict when context offers few clues? Across Llama, Qwen, Gemma, and Pythia, we find a single direction in the unembedding that encodes a unigram prior—the frequencies of individual tokens in training text. The model relies less on this prior as context becomes more informative; adjusting its strength at inference time can steer predictions and improve mathematical reasoning in Llama-3.2-3B-Instruct.

Panel a: Llama-3.2-3B token embeddings colored by frequency, with an arrow marking the prior direction.
Panel a: the token-frequency gradient in Llama-3.2-3B's unembedding. The arrow marks the prior direction.

Jacobian Scopes for Causal Attribution

When an LLM answers, which prompt tokens drove that decision? Jacobian Scopes attribute predictions—logits, full distributions, and uncertainty—to inputs via fast vector-Jacobian products, with strong performance on attribution benchmarks.

Jacobian Scope visualization: per-token influence on predicting liberal versus conservative from the same prompt.
Semantic Scope reveals the potential for political bias that may be present in models such as LLaMA3.2. The prediction for the subject being a “liberal” is attributed to the input token “Columbia”, while “conservative” to the token pairs for “the South”.

LLMs think like a kernel machine

How do LLMs learn from in-context data? Using Intensive Principal Component Analysis, we visualize their density-estimation trajectories and find an inference process akin to kernel regression, with a kernel whose width and shape adapt as evidence accumulates.

Rotating inPCA comparison of LLaMA-2 70B, Gaussian KDE, and Bayesian histogram density-estimation trajectories. Legend for the density-estimation trajectories.
In-context density-estimation trajectories of LLaMA-2 70B, Gaussian KDE, and Bayesian histogram, shown in the first three inPCA dimensions.

Memory–abstraction trade-off in diffusion models

What separates abstract thoughts from concrete memories? Treating memories as fixed points, we cast abstraction as a hierarchy of slow manifolds and build a diffusion RNN that extracts them from data—with an analytic theory linking data distributions to the learned manifolds.

Diffusion RNN extracts a hierarchy of slow manifolds (red) from dimensionally reduced MNIST data.

In-context neural scaling law

Can a pretrained LLM make sense of an unseen synthetic language—say, symbols from a random Markov process? LLaMA 2 models in-context learn a range of stochastic systems, physical and symbolic, revealing an in-context neural scaling law.

Noisy logistic-map states, next-state prediction given the current state, and aggregate in-context loss.

Embedding and attention in non-Euclidean spaces

Hyperbolic volume grows exponentially with radius—like the branching of a tree—making it a natural host for hierarchical data. Shadow Cones embed partial orders by casting nested shadows: a child lies in its parent’s cone, yielding a fast algorithm for embedding concept graphs.