Skip to main content
Search

# Sample Complexity

8 related articles

Subscribe to updates
  1. arXiv

    KL-regularized RL gets a sharp sample count: with a generative model, samples scale linearly in the effective horizon and temperature no longer slows learning

    KL-regularized reinforcement learning, the objective behind maximum-entropy RL and RL fine-tuning of language models from human feedback, previously had guarantees whose sample counts grew with a worse power of the effective horizon, and no MDP lower bound depended on the temperature. For tabular discounted MDPs with a generative model, this work shows that a plug-in estimator, which draws a fixed number of samples per state-action pair and solves the empirical MDP, reaches KL suboptimality at the temperature scale with samples linear in the effective horizon, and that no algorithm, even one choosing where to sample, can do with fewer.

  2. arXiv

    Robust Q-learning drops the generative model: one Markovian trajectory suffices, with near-minimax sample complexity

    For robust Markov decision processes with total-variation uncertainty sets, a model-free asynchronous Q-learning algorithm relies only on a single Markovian trajectory from a behavior policy, without a generative model or queries to arbitrary state-action pairs. It estimates robust Bellman updates through a resampled surrogate next state, uses $|S||A|$ space, and reaches entrywise ϵ\epsilon accuracy with a sample complexity of order ∣S∣∣A∣tmixϵ2μmin⁡\frac{|S||A| t_{\mathrm{mix}}}{\epsilon^2 \mu_{\min}} up to logarithmic factors, carrying the extra factor inherent to Q-learning.

  3. arXiv

    How much of a policy's value comes from reacting to state: a covariance identity isolates adaptivity, but deterministic policies cannot be audited from their own logs

    When a regulator evaluates a learned policy, the inputs and outputs are usually all that is visible, so it is hard to tell whether value comes from acting on the state or from average behavior. This work shows that, in a finite-horizon Markov decision process, the expected-cost difference between a policy and its open-loop marginal equals exactly the covariance between state and action, and it bounds adaptive value by the mutual information the policy uses. Auditing that quantity under bandit feedback needs a sample size inversely proportional to an average KL divergence, while a deterministic non-constant policy is not identifiable from its own logs.

  4. arXiv

    A test set may already be in the training data: accurate evaluation is possible only when the distribution concentrates on few points

    Large-model training corpora often cover public benchmarks, so evaluators can no longer assume a fresh test set. This work formalizes evaluability without assuming the test set was held out and without constraining the model or its training algorithm, showing that accurate evaluation from model outputs on the test set alone is feasible if and only if the distribution is close to small support, with sample-complexity bounds tight up to logarithmic factors; allowing extra queries to the model, via self-correctors from program testing, widens the evaluable distributions.

  5. arXiv

    Transferable rewards without a flood of expert demos: actively picking dynamics cuts the demonstrations needed

    Inverse reinforcement learning aims to recover expert rewards that transfer across environments with differing dynamics, and expert demonstrations are costly, so choosing which dynamics to collect them under matters. A two-phase approach first cycles through the available dynamics to obtain a reliable reward estimate, then actively selects dynamics to reduce uncertainty about the expert reward, yielding sample-complexity guarantees on transfer that hold for any dynamics with a rate independent of the number of available dynamics. Empirically, the method significantly reduces the number of expert demonstrations needed to learn a transferable reward.

  6. arXiv

    AI agents vote down their own run: a diamond-spin lab rejects a passing experiment, then earns the evidence

    A laboratory of large-language-model agents, each holding a separate research role, asked how many calibration measurements are needed to predict a diamond nitrogen-vacancy spin ensemble's response to arbitrary phase-controlled microwave pulse sequences. The principal-investigator agent tightened a proposed bound of 91 calibrations to an exact minimum of 10; the first acquisition met all ten precommitted accuracy criteria but failed phase and reference controls and was ruled inconclusive, and after the agents revised the protocol without relaxing the criteria, the final measurement supported the prediction while two interpolation methods predicted zero.

  7. arXiv

    Probability measures get sparse dictionaries: embed once by optimal transport, then train a plain autoencoder

    Embedding probability measures into a Hilbert space before learning a sparse dictionary turns training from repeated Wasserstein distance solves into one autoencoder optimization, cutting reconstruction loss by roughly two to three orders of magnitude against Wasserstein dictionary learning under the same time budget.

  8. arXiv

    Masked diffusion models no longer pay for sequence length: latent structure sets the sample count

    Transformer-based masked diffusion models now have an end-to-end statistical guarantee: when the data distribution is well approximated by a mixture of hidden Markov trees, the training samples needed for a target KL accuracy are governed by latent structural complexity rather than growing exponentially with the sequence space. The analysis covers both the parallelization error of parallel unmasking and the finite-sample estimation error of the mask predictor, and the transformer architecture and sampling procedure need no knowledge of tree topologies or component parameters, adapting to unknown latent structure.

You’re all caught up