跳转至

Foundation models by topic

简体中文 · English

Browse research areas and method families across architecture, pretraining, multimodality, and inference efficiency. Post-training algorithms have their own LLM post-training catalog.

Model architecture

JEPA and decision representations

  • GUI-JEPA: Candidate-Conditioned JEPA Latents as a Fast Action Policy for Mobile GUI Agents (gui-jepa): The policy does not equate latent proximity to a goal screenshot with action value. Its predictor is trained on actual candidate sets with a listwise decision loss; the structural head sees candidate visual information through the conditioned latent. Bounded component only; full experiments are not reproduced.
  • SWIM: Learning What Matters from Behaviors for JEPA World Models (swim): Stage 1 gates only the future displacement, leaving the entire current state visible. Stage 2 scales previously standardized states by the square root of the frozen weights and matches latent and state distances for the same frame pairs. Bounded component only; full experiments are not reproduced.
  • DT-JEPA: Learning to Predict Future Representations for Decision-Making (dt-jepa): A reparameterized sample conditions the Decision Transformer via AdaLN with separate scale and shift MLPs. Offline training combines behavior cloning and NLL; online adaptation uses the smaller of two target critics for a multi-step target and a Q1-guided actor objective. Bounded component only; full experiments are not reproduced.

JEPA and relational representations

JEPA and geometric representations

  • ALPS: Visual State Encoders for World Models (alps): A four-layer predictor uses geometry to predict clean crop B teacher tokens from enhanced crop A. Supervise only B centers in A; discard the predictor and retain the student and language interface. Bounded component only; full experiments are not reproduced.
  • LocaLeJEPA: Maximum Entropy Representations under Geometric Priors (locale-jepa): Retain clusters of size at least four; if none qualify, use the whole batch and record fallback. Cluster centers need not approach one common origin. Bounded component only; full experiments are not reproduced.

Recurrent memory and sequence mixing

  • Kronecker Delta Rule for Long Context Linear Attention (kronlstm): In vector form \(a=v\otimes k\), the step is \(s'=(I-\beta aa^T)s+\beta ca\). Overlap depends on both key and value inner products; orthogonality of either preserves that association for the current update, not arbitrary long-sequence retention. Bounded component only; full experiments are not reproduced.

On-device multimodal agent models

  • Muse Glimmer 30B: An On-Device Agentic Model (muse-glimmer-30b): Training starts with distillation from Muse Spark logits. Near the end of pretraining, it switches to cross-entropy to address top-\(k\) truncation bias. Not reproduced due to cost (review complete); research material only, no executable entry.

Diffusion and generative models

MoE and dynamic routing

Conditional memory and knowledge injection

MoE, state space, and residual paths

Recurrence and elastic depth

Dynamic layer routing and residual paths

  • MACRO: Markov Chain Routing of Transformer Layers (macro): MACRO represents skips, repeats, and residual additions with a context-conditioned Markov policy. Feedback updates the routing distribution and top-k Viterbi decodes candidate programs without changing base-model weights.

Recurrence and latent computation

Generative experts and robust gating

Mechanistic interpretation and topic models

  • Model Directions, Not Words: Mechanistic Topic Models Using Sparse Autoencoders (mechanistic-etm): Eqs. 16–19 use independent SAE decoder columns only for steering: combine and normalize those directions, remove the existing parallel component, then inject the specified strength. A negative control with different encoders and decoders catches implicit weight tying. Bounded component only; full experiments are not reproduced.

Attention and long context

Multi-state linear attention

  • Multi-State Linear Attention with Generalized Second-Order Routing (merlin): This changes segment aggregation; it does not replace the recurrent memory with the moment statistics or recover exact full softmax attention. The second-order cumulant analysis assumes equal-length segments. Bounded component only; full experiments are not reproduced.

Query-conditioned test-time training

Attention kernels and backpropagation

Sparse, gated, and dynamic attention

Sequence compression and differentiable test-time regression

  • Rethinking Sequence Compression with Test-time Regression (ttr-compressor): The outer task trains projection and regularizer through the exact solve. Test-time refitting changes input-dependent coefficients, not model weights; the base has \(2d\) trainable parameters. Bounded component only; full experiments are not reproduced.

RoPE and native dimension compression

Sparse sampling and estimation

Sparse attention and video generation

Linear and separable attention

KV cache and context compression

Linear attention and state space

Task-guided KV cache compression

Positional encoding and KV compression

Pretraining and data

Optimizers and parameter-efficient fine-tuning

Distillation theory and finite policy learning

Optimizers and training stability

Training frameworks and composable experiments

Byte-level modeling and causal chunking

Self-supervised representations and regularization

  • Dataset-Level Regularization for Joint-Embedding Self-Supervised Learning (oob-regularization): The Gaussian variant projects onto random unit directions, sorts the examples along each direction, and matches standard-normal quantiles. The contrastive variant optimizes uniformity on the unit sphere while preserving the correspondence with each current representation. Bounded component only; full experiments are not reproduced.

Data selection and quality estimation

Diffusion language model training

  • On Trajectory-Aware Training for Masked Diffusion Language Models (pumba): During training, unroll consecutive denoising steps along the model's own progressive-unmasking trajectory. Pass a hidden carry to the next step and backpropagate through time within a fixed window, so an earlier step learns to produce a useful carry for later steps.

Latent and multi-step prediction

Data cleaning, orchestration, and selection

Continual learning and memory consolidation

Enterprise data and domain adaptation

  • Customizing an LLM for Enterprise Software Engineering (gemini-for-google): Deduplication, quality filtering, and contamination controls produce mid-training data. General-data replay limits catastrophic forgetting, followed by post-training for internal tools and deployment evaluation.

Coordinated pretraining and post-training

Optimizers and training efficiency

  • Muon is Scalable for LLM Training (muon): The paper adds weight decay and shape-aware scaling for large-scale training. Parameters outside hidden matrices still use AdamW.

Multimodal foundation models

Word–region alignment

Medical imaging and structured visual decisions

  • PathJev: Decision-Native Vision–Language Modelling for Histopathology (pathjev): The head and two-stage training follow OmniMed-Jev. The pathology contribution is the 40K decision states, typed labeling, patient/slide and overlap audit, unseen-organ tests and generative readout comparisons. Not reproduced due to cost (review complete); research material only, no executable entry.

World models and embodied control

  • Persistent Planar Memory for Video World Models (ppmem): Each local plane has seven parameters: center depth, two rectangle scales, and a quaternion. Planes are merged after transformation to world coordinates. Not reproduced due to cost (review complete); research material only, no executable entry.
  • Gen4P: Video Generators are 4D Perceivers (gen4p): Gen4P does not recover geometry through repeated denoising: it encodes clean video latents with one DiT pass and uses on-demand cross-attention queries to output 3D tracks, point clouds, depth, and camera parameters. Continuous-time queries support different lengths, frozen depth LoRA supplies an additional geometry prior, and separate 4D LoRA learns the new tasks. Not reproduced due to cost (review complete); research material only, no executable entry.
  • EgoJEPA: Beyond Retargeting for Dexterous Manipulation from Egocentric Videos (egojepa): Hand movement in human first-person video need not be mapped with hand-written rules to each robot's actions. EgoJEPA first conditions JEPA on hand-motion differences to predict future visual representations, then jointly models vision and latent actions, and finally adapts only the target robot's input and output action interfaces. Not reproduced due to cost (review complete); research material only, no executable entry.
  • One-Step Prediction Fidelity Is a Poor Proxy for Planning Performance in JEPA World Models (jepa-prediction-planning): Planning is evaluated separately with matching checkpoints, tasks and paired seeds. Success rates cannot be inferred from prediction metrics. Evaluation protocol interface; not a model capability result.
  • SmolWAM: A Latent World-Action Model for Robotic Manipulation on Edge Devices (smolwam): A single-frame Wan VAE and 0.5M adapter map back to the clip latent space. Cached BERT features are padded to 512 positions, retaining a zero-bias attention sink.
  • LeWAM: End-to-End World and Action Modeling with JEPAs (lewam): The visual encoder combines a ResNet-18 trained from scratch, spatial softmax, and a two-layer projection MLP. The simulation model has about 16.9M parameters, latent/MoT width 192, 4 layers and 4 heads.
  • V-JEPA Policy: Building Effective World-Action Models on Predictive Visual Latents (vjepa-policy): Freeze V-JEPA 2.1 vision encoding and jointly train an instruction-conditioned future-latent predictor and flow-matching action expert. Action generation reads the predicted future context.

Recurrent visual encoding and adaptive compute

  • AdaLoop: Thinking with the Vision Encoder (adaloop-vision): AdaLoop freezes pretrained visual weights, pools question tokens to modulate shared visual blocks, and repeatedly updates the visual state. The question can change how many passes the model uses rather than leaving all reasoning to the language decoder. Bounded component only; full experiments are not reproduced.
  • Retrofitting Recurrent Vision-Language Models from Pretrained Backbones (rr-vlm): RR-VLM fuses prelude context and recurrent state with a linear adapter, then performs text recovery, projector alignment, and full-model instruction recovery. Text and final recovery also distill from the original unpruned teacher. Bounded component only; full experiments are not reproduced.

Role collaboration and policy distillation

Vision–language adaptation and representations

Spatial understanding and structured visual reasoning

  • MJev-Spatial: System One Spatial Intelligence via Native Attention Matching (mjev-spatial): The decision marker supplies the query, and candidate-end markers supply keys using the backbone's native final attention projections. Continuous values use bins or three-stage regional refinement, without a separate classification head. Bounded component only; full experiments are not reproduced.

Geometrically consistent novel-view generation

JEPA and physical-state supervision

Video representations and elastic cross-device inference

  • EV-JEPA: Elastic V-JEPA for Any Device and Cross-Device Inference (ev-jepa): EV-JEPA shares parameters across nested widths, jointly training narrow, wide, and narrow-to-wide paths at every step. A frozen video teacher constrains their common representation space, and adjustable token compression supports transfer. Not reproduced for now due to cost; validation pending, no executable entry.

Natural-language decision encoders

Video understanding and token compression

Vision–language adaptation and efficient inference

Visual tokens and cross-modal retrieval

Visual generation and understanding

Verifiable visual reasoning and evaluation

Contrastive pretraining and self-distillation

Generative auxiliary supervision for better understanding

Mobility trajectories and place representations

Efficient visual token compression

  • SmolVLM: Redefining small and efficient multimodal models (smolvlm): Move neighboring spatial tokens into channels with pixel shuffle, then project into LM space with an MLP. The paper also studies long contexts, image tiling, learned position tokens and data mixtures for 256M/500M/2.2B image/video models.

Cross-modal connectors and frozen backbones

Inference and systems efficiency

Compiler correctness and hierarchical fault isolation

  • XLA from the Trenches: Correctness on TPU Workloads at Scale (xla-trenches): Isolation proceeds from a large HLO graph to a fusion, then inspects intermediate VREG values without splitting the fusion. This is compiler-correctness infrastructure, not a new recommender or an industrial recommendation paper passing an A/B gate. Not reproduced due to cost (review complete); research material only, no executable entry.

KV cache and context compression

  • Folding Attention Biases into Keys Makes Compacted KV Caches Run on Standard Kernels (fold-bias): This is an approximation, not an exact replacement for all future queries: a dot product is linear in the query, so it cannot represent a nonzero constant over the entire query space. The fit constrains the mean projection to one on the calibration queries and limits its variation. Bounded component only; full experiments are not reproduced.

Quantization and residual compensation

System One, JEV, and structured decisions

Auditable detection and activation interpretation

  • Building Auditable AI Text Detectors with Sparse Autoencoders (detector-scope): A sparse linear head yields logit contributions verifiable by zeroing features. Neither a text description nor a rank score certifies concept semantics, human review or low false-positive risk. Bounded component only; full experiments are not reproduced.

MoE routing and KV cache

  • MoEZip: Routing-Aware KV Cache Compression for Sparse Mixture-of-Experts LLMs (moezip): Use Fisher trace from prefill instruction-token gates as a sensitivity proxy and propagate it to context locations with headwise maximum attention. Retain global routing anchors in all layer/heads and fill remaining budget using external base-compressor scores, without future answer tokens. Bounded component only; full experiments are not reproduced.

Quantization

System One / JEV retrieval evaluation protocols

Multilingual evaluation and scoring protocols

Multi-vector retrieval and fixed-budget pooling

  • FLINT: Fixed-budget Late Interaction (flint): FLINT uses separate learned query/document latent sets after a shared backbone, with shared cross-attention and FFN to emit fixed \(V\) vectors per side. This is neither first-token truncation nor averaging. Bounded component only; full experiments are not reproduced.

Speculative decoding and KV caching

Structured decisions and adaptive inference

  • Prism-1: Position- and Rival-Independent Scoring for System One Models (prism-1): A softmax combines independent scores. Softmax alone cannot guarantee this without attention isolation and position reset. Bounded component only; full experiments are not reproduced.
  • Selective Reasoning for Typed Decision Models: When to Start, Read, and Stop (selective-reasoning): The controller separates three decisions: a benefit gate decides whether to begin reasoning; a trigger from the natural generation state decides when to request explicit answer readout; a normalized-entropy threshold decides whether to accept a candidate and cancel further generation. The original online system also has concurrent generation and reading, request cancellation, exact prefix restoration, and cost accounting. Evaluation protocol interface; not a model capability result.
  • CoTS: Competitive Token Selection for Latent Reasoning in Large Language Models (cots): CoTS uses top-two probability competition to control soft mixing, retains at least half the sampled token's weight, and irreversibly switches to discrete output after reasoning. See Sections 3.1–3.3, Equations 1–6 and Section 4 for experiments and ablations. Bounded component only; full experiments are not reproduced.

Diffusion decoding and caching

Parallel serving and KV caching

State caching and low-precision replay

  • ReplaySSM: Cache SSM Inputs, Not State (replayssm): The acceptance length advances only the committed-input pointer; rejected drafts never enter the recurrence. Separating a low-precision copy used for output reads from a high-precision recurrence anchor prevents repeated feedback of read quantization error.

Dynamic compute and model compression

Capability evaluation and inference protocols

Retrieval and reranking calibration

Agent inference and KV caching

Model routing and dynamic compute

In-context learning and adaptive inference

Safety structure and activation diagnostics

Structural pruning and reversible compression

Multilingual knowledge exploration

  • Cross-Lingual Exploration for Parametric Knowledge (cross-lingual-exploration): Gemini judges semantic correctness and cross-language consistency. A counter or string translator would break the defining mechanism, so no heuristic component is invented. Not reproduced due to cost (review complete); research material only, no executable entry.

Multilingual algorithm evaluation protocols

  • XLGoBench: Detecting cross-lingual skill gaps with algorithmic tasks (xlgobench): Graph-SP requires a unique shortest path, and T-Rank uses a Hamiltonian DAG with transitive edges and a 1:3 node-to-edge ratio. The local callback receives deep copies of allowlisted task/problem fields; exact JSON scoring needs no LLM judge. Bounded component only; full experiments are not reproduced.

Active evaluation and performance estimation

Verifiers and result integration

  • FUSE: Ensembling Verifiers with Zero Labeled Data (fuse-verifiers): FUSE reads only candidate score matrices, estimates query-dependent verifier quality from second and third moments, selects transforms to reduce dependence, constructs correctness posteriors, and optimizes an ensemble. True labels are used only for independent evaluation. Not reproduced due to cost (review complete); research material only, no executable entry.

Generative retrieval and structured decisions

Generative watermarking and edit detection