arXiv Statistics ML By Xinwei Shen, Zijian Guo, Francis Bach

Generalized Engression Models

Read the original on arXiv Statistics ML →

The paper introduces generalized engression models, a unified nonparametric framework for estimating the full conditional distribution of multivariate outcomes that may include continuous, binary, categorical, ordinal, or ranking variables. Building on the engression deep generative model, it incorporates data‑type‑specific link functions and a stochastic perturbation to smooth the loss, allowing gradient‑based training even with discontinuous links. The authors prove universal representation results for continuous, discrete, and mixed outcomes, and demonstrate that the method matches or surpasses existing type‑specific and joint species distribution models in simulations and two real‑world applications.

Machine-generated by The Flow from the publisher's headline and feed description — not written or checked by a human. The full article lives at arXiv Statistics ML.

arXiv Machine Learning
Aug 24

Conditional-Independence-Regularized Distributional Autoencoders for Mixed-Type Data

The paper introduces Conditional-Independence-Regularized Distributional Autoencoders, a framework for learning low-dimensional representations of mixed-type data that includes both numerical and categorical variables. It uses an energy-score objective for numerical variables, a likelihood objective for categorical variables, and an auxiliary conditional independence regularization term to capture dependencies between variable types. The authors provide theoretical analysis and demonstrate that the method improves categorical distribution recovery, achieves competitive overall conditional distribution recovery, and preserves mixed-type dependence structure on synthetic and real-world datasets.

By Siyuan Tang, Gongjun Xu, Ji Zhu
arXiv Machine Learning
Sep 25

Sufficiently Reduced Distributional Regression

Sufficiently Reduced Distributional Regression (SRDR) is a generative approach that merges conditional distribution estimation with nonlinear sufficient dimension reduction (SDR). By framing SDR as a risk minimization problem using strictly proper scoring rules, SRDR jointly learns a dimension reduction map and a generative prediction model through minimization of the energy score, which can be estimated via sampling. The method extends to multi‑environment data and classification, and theoretical results show convergence of estimated conditional distributions in energy distance, implying asymptotic sufficiency. In experiments on CT slice localization, superconductivity data, and digit classification, SRDR recovers low‑dimensional sufficient structure and matches or surpasses state‑of‑the‑art nonlinear SDR methods in representation quality and predictive performance.

By Alexander Henzi, Tiange Liu, Xinwei Shen
Hugging Face Trending Papers
Jul 5

Optimal Mixture-of-Experts Model Averaging for Conditional Generative Models

Conditional generative models have emerged as powerful tools for sampling from target conditional distributions, driving substantial advances across a wide range of scientific and applied domains. As these models proliferate, practitioners often face multiple plausible generators whose performance can vary with the task, data, or input condition.

arXiv Machine Learning
3d ago

Survival In-Context: Amortized Bayesian Survival Analysis via Prior-Fitted Networks

The paper introduces Survival In-Context (SIC), a prior‑fitted in‑context learning model for survival analysis that is pretrained solely on synthetic data generated from a flexible survival prior. SIC approximates Bayesian posterior predictive inference, allowing individualized survival predictions in a single forward pass without task‑specific training or hyperparameter tuning. Experiments on real‑world survival datasets show that SIC matches or outperforms classical and deep survival models, especially when data are limited.

By Dmitrii Seletkov, Paul Hager, Georgios Kaissis, Rickmer Braren, Daniel Rueckert, Raphael Rehms