ContextFlow++: Generalist-Specialist Flow-based Generative Models with Mixed-Variable Context Encoding

• URL: http://arxiv.org/abs/2406.00578v1
• Date: Sun, 2 Jun 2024 00:00:00 GMT
• Title: ContextFlow++: Generalist-Specialist Flow-based Generative Models with Mixed-Variable Context Encoding
• Abstract summary: We propose ContextFlow++ to overcome limitations using an additive conditioning with explicit generalist-specialist knowledge decoupling. Our experiments on rotated MNIST-R, corrupted CIFAR-10C, real-world ATM predictive maintenance and SMAP unsupervised anomaly detection benchmarks show that the proposed ContextFlow++ offers faster stable training and achieves higher performance metrics.
• Score: 0.8192907805418583
• License: http://creativecommons.org/licenses/by/4.0/
• Abstract: Normalizing flow-based generative models have been widely used in applications where the exact density estimation is of major importance. Recent research proposes numerous methods to improve their expressivity. However, conditioning on a context is largely overlooked area in the bijective flow research. Conventional conditioning with the vector concatenation is limited to only a few flow types. More importantly, this approach cannot support a practical setup where a set of context-conditioned (specialist) models are trained with the fixed pretrained general-knowledge (generalist) model. We propose ContextFlow++ approach to overcome these limitations using an additive conditioning with explicit generalist-specialist knowledge decoupling. Furthermore, we support discrete contexts by the proposed mixed-variable architecture with context encoders. Particularly, our context encoder for discrete variables is a surjective flow from which the context-conditioned continuous variables are sampled. Our experiments on rotated MNIST-R, corrupted CIFAR-10C, real-world ATM predictive maintenance and SMAP unsupervised anomaly detection benchmarks show that the proposed ContextFlow++ offers faster stable training and achieves higher performance metrics. Our code is publicly available at https://github.com/gudovskiy/contextflow.

Related papers

• ContextFlow: In-Context Flow Matching for Robot Manipulation [91.60992853468242]
  We introduce ContextFlow, a conditional flow-matching model that learns continuous action distributions for in-context imitation learning.<n>On LIBERO, ContextFlow outperforms ICRT by 35 percentage points in average success rate on unseen task configurations.<n>On real robots, it generalizes to unseen configurations of both single-arm and bimanual tasks, achieving 40% success on a new pen-uncapping configuration.
  arXiv  Detail & Related papers  (2026-09-06T21:53:58Z)
• ConFlow: Constraints-Guided Learning with Flow Matching for Motion Generation [2.0175723444555307]
  In robot motion generation we often have additional constraints that might not be present in the collected data.<n>We propose ConFlow, a constraint-guided flow matching framework that incorporates constraint information directly into the training objective.<n>Experiments on a two-robot navigation task demonstrate that ConFlow achieves lower collision rates and higher trajectory quality.
  arXiv  Detail & Related papers  (2026-07-15T23:27:04Z)
• Bounded Context Management for Tabular Foundation Models on Stream Learning [8.331524801839162]
  We propose a future information view that yields three practical requirements for context management.<n>We instantiate these requirements as CURE, a context-managing policy with entropy-gated admission and redundancy-aware eviction.<n>CURE shows up to 27.0% relative improvement over classical stream learners, remains robust across multiple TFM backbones, and ranks first among other policy variants.
  arXiv  Detail & Related papers  (2026-06-17T04:24:26Z)
• Contexting as Recommendation: Evolutionary Collaborative Filtering for Context Engineering [37.86884288597418]
  Large Language Models (LLMs) are highly sensitive to their input contexts.<n>Existing methods predominantly treat this as a global search problem, seeking a single context strategy.<n>We introduce textbfNeural Collaborative Context Engineering (NCCE), a framework that transitions optimization from a static global search to dynamic, instance-wise routing.
  arXiv  Detail & Related papers  (2026-05-15T08:17:21Z)
• Predicting Task Performance with Context-aware Scaling Laws [56.6850444554434]
  We propose a straightforward, interpretable framework that jointly models downstream performance as a function of the training compute and the provided context.<n>We empirically validate our framework by fitting it on the observed downstream performance of extended-context variants of Llama-2-7B and Llama-2-13B.<n>Our results demonstrate that our framework accurately models in-distribution downstream performance, generalizes across three orders of magnitude in training compute, and reliably extrapolates performance as the amount of context increases.
  arXiv  Detail & Related papers  (2025-10-16T17:35:18Z)
• PICASO: Permutation-Invariant Context Composition with State Space Models [98.91198288025117]
  State Space Models (SSMs) offer a promising solution by allowing a database of contexts to be mapped onto fixed-dimensional states. We propose a simple mathematical relation derived from SSM dynamics to compose multiple states into one that efficiently approximates the effect of concatenating raw context tokens. We evaluate our resulting method on WikiText and MSMARCO in both zero-shot and fine-tuned settings, and show that we can match the strongest performing baseline while enjoying on average 5.4x speedup.
  arXiv  Detail & Related papers  (2025-02-24T19:48:00Z)
• Can Transformers Learn Full Bayesian Inference in Context? [13.479322264788367]
  In this paper, we demonstrate that transformers can perform full Bayesian inference for commonly used statistical models in context.<n>We introduce a general framework that builds on ideas from prior fitted networks and continuous normalizing flows.<n>Experiments on real-world datasets demonstrate that our ICL approach yields posterior samples that are similar in quality to state-of-the-art MCMC or variational inference methods.
  arXiv  Detail & Related papers  (2025-01-28T10:04:53Z)
• A Case Study on Context-Aware Neural Machine Translation with Multi-Task Learning [49.62044186504516]
  In document-level neural machine translation (DocNMT), multi-encoder approaches are common in encoding context and source sentences. Recent studies have shown that the context encoder generates noise and makes the model robust to the choice of context. This paper further investigates this observation by explicitly modelling context encoding through multi-task learning (MTL) to make the model sensitive to the choice of context.
  arXiv  Detail & Related papers  (2024-07-03T12:50:49Z)
• Online Joint Fine-tuning of Multi-Agent Flows [12.851745991007169]
  I describe a procedure for online joint fine-tuning of an entire flow inspired by the Learning to Search framework. The approach leverages simulator access to reduce preferences over entire episodes to preferences over individual node outputs. I apply to the multi-hop QA dataset Musique achieving a state-of-the-art result.
  arXiv  Detail & Related papers  (2024-06-06T21:21:03Z)
• Guided Flows for Generative Modeling and Decision Making [55.42634941614435]
  We show that Guided Flows significantly improves the sample quality in conditional image generation and zero-shot text synthesis-to-speech. Notably, we are first to apply flow models for plan generation in the offline reinforcement learning setting ax speedup in compared to diffusion models.
  arXiv  Detail & Related papers  (2023-11-22T15:07:59Z)
• Coherent Entity Disambiguation via Modeling Topic and Categorical Dependency [87.16283281290053]
  Previous entity disambiguation (ED) methods adopt a discriminative paradigm, where prediction is made based on matching scores between mention context and candidate entities. We propose CoherentED, an ED system equipped with novel designs aimed at enhancing the coherence of entity predictions. We achieve new state-of-the-art results on popular ED benchmarks, with an average improvement of 1.3 F1 points.
  arXiv  Detail & Related papers  (2023-11-06T16:40:13Z)
• Disentanglement via Latent Quantization [60.37109712033694]
  In this work, we construct an inductive bias towards encoding to and decoding from an organized latent space. We demonstrate the broad applicability of this approach by adding it to both basic data-re (vanilla autoencoder) and latent-reconstructing (InfoGAN) generative models.
  arXiv  Detail & Related papers  (2023-05-28T06:30:29Z)
• Context-aware Fine-tuning of Self-supervised Speech Models [56.95389222319555]
  We study the use of context, i.e., surrounding segments, during fine-tuning. We propose a new approach called context-aware fine-tuning. We evaluate the proposed approach using the SLUE and Libri-light benchmarks for several downstream tasks.
  arXiv  Detail & Related papers  (2022-12-16T15:46:15Z)
• VQ-Flows: Vector Quantized Local Normalizing Flows [2.7998963147546148]
  We introduce a novel statistical framework for learning a mixture of local normalizing flows as "chart maps" over a data manifold. Our framework augments the expressivity of recent approaches while preserving the signature property of normalizing flows, that they admit exact density evaluation.
  arXiv  Detail & Related papers  (2022-03-22T09:22:18Z)
• Attentive Contractive Flow with Lipschitz-constrained Self-Attention [25.84621883831624]
  We introduce a novel approach called Attentive Contractive Flow (ACF) ACF utilizes a special category of flow-based generative models - contractive flows. We demonstrate that ACF can be introduced into a variety of state of the art flow models in a plug-and-play manner.
  arXiv  Detail & Related papers  (2021-09-24T18:02:49Z)
• Measuring and Increasing Context Usage in Context-Aware Machine Translation [64.5726087590283]
  We introduce a new metric, conditional cross-mutual information, to quantify the usage of context by machine translation models. We then introduce a new, simple training method, context-aware word dropout, to increase the usage of context by context-aware models.
  arXiv  Detail & Related papers  (2021-05-07T19:55:35Z)

This list is automatically generated from the titles and abstracts of the papers in this site.

This site does not guarantee the quality of this site (including all information) and is not responsible for any consequences.