Skip to content

Category

machine learning

1,973 papers

#machine learning Preprint Jul 2026

CHM-Net: Center Heatmap-driven Macro-Micro Modeling Network for MRI-based Microbial Density Stratification

This work investigates MRI-based Microbial Density Stratification as a patient-level representation learning task, and Center Heatmap-driven Macro-micro modeling Network (CHM-Net) is introduced for this task, establishing the link between imaging phenotypes and microbial states through center heatmap-guided small-lesion response localization.

Jiaming Liang, Hao Chen, Ting Li et al. · 0 citations
#artificial intelligence Preprint Open access Aug 2026

Turning Off-Policy Tokens On-Policy: A Plug-in Approach for Improving LLM Alignment

Reinforcement learning (RL) post-training for large language models (LLMs) follows a efficient paradigm of "rollout then update", which inevitably results in off-policy training data. To resolve this, Importance sampling (IS) is proposed, while the token-level ratios compound over long sequences, causing severe variance exploded. A natural idea is "transferring" these off-policy token into on-policy token, so that the importance scores for correction are unnecessary. Following this idea, we propose Selective Importance Sampling (SIS), which is inspired by rejection sampling. Concretely, SIS implements by viewing off-policy model as proposal distribution, and implement a token-level rejection test: accepted tokens are viewed as on-policy, so that receive unit importance score, while rejected tokens retain the standard IS correction. Our proposed SIS is theoretically proved reducing the gap between token-level and sequence-level off-policy gradient estimators. The SIS acts as a plug-in that only modifies the importance ratio in the policy loss, adding negligible wall-clock overhead, and can be combine with a vast vary of RL post-training algorithms. Experiments on dense and MoE LLMs across math and agent benchmarks show that SIS consistently improves all objectives, while providing substantially stronger robustness under off-policy data.

Yu Li, Xiuyu Li, Mingyang Yi et al. · 0 citations
#machine learning Preprint Open access Aug 2026

Spectrally Safe Neural Operator Warm-Starts for Large-Scale Newton Solvers

Neural operators are increasingly used to warm-start Newton solvers for nonlinear PDEs, on the premise that a low test error places the initial guess inside the basin of attraction. We show that this premise is unreliable. An operator trained to the relative \(L^2\) error \(O(10^{-3})\) can still produce an initial state in which the discrete Jacobian is indefinite, because the mean-squared training controls error on average while leaving localized pointwise violations of the underlying physics. For a nearly incompressible hyperelasticity problem, we trace this to the predicted volume change: the operator disperses \(\mathrm{det} F\) well away from one, and the resulting Jacobian acquires negative eigenvalues even when the predicted field is visually indistinguishable from the reference. At a small scale, this is a nuisance; at a multi-million degree-of-freedom scale, it is disqualifying, since the conjugate gradient and other Krylov solvers needed for memory-feasible Newton steps assume a definite spectrum. We then show that a short, label-free fine-tuning phase -- penalizing the operator against the discrete energy, with no additional solution data -- shifts the Jacobian spectrum back to positive definite. Combined with an inexact outer loop, this gives a warm-started Newton method that converges across the full loading range where the unregularized operator fails, reaching up to 5.4\(\times\) wall-clock speedup over incremental continuation on a 3D problem with 6.4 million degrees of freedom.

Jaemin Oh, Youngkyu Lee, Jerome Darbon et al. · 0 citations

SEAM: Shortcut-Aware Real-Time Detection of Scripted vs. Spontaneous Speech for Interview Guardrails

SEAM is presented, a shortcut-aware framework for real-time scriptedness detection that combines uniform preprocessing, seam-aware sampling, non-speech augmentation, and a compact DistilHuBERT backbone that demonstrates that robust real-time scriptedness detection depends not only on the backbone, but on shortcut-aware data design and evaluation.

Vsevolod Kovalev, Pranay Manocha · 0 citations

Memory by Design: Probabilistic Sequence Layers

Distilling Bayesian Layers into a pretrained 340M Gated DeltaNet improves RULER long-context retrieval over a matched-compute control, and improves robustness beyond the training regime in controlled collision studies, learned associative recall, and the Zoology MQAR benchmark.

Matthew Dowling, H. Jeon, Cristina Savin et al. · 0 citations

Adaptive AI Task Partitioning and Safe Offloading in Heterogeneous Edge-Cloud Continuum

A framework that dynamically splits neural network layers across the heterogeneous continuum and achieves reductions in energy and end-to-end latency is proposed, confirming the superiority of adaptive to static partitioning.

Akuen Akoi Deng, Eimantas Butkus, Alfreds Lapkovskis et al. · 0 citations
#machine learning Open access May 2026

Nonlinear GENERIC-Embedded Neural Networks (N-GENNs): Learning GENERIC dynamics with non-quadratic dissipation potentials

Nonlinear GENERIC-Embedded Neural Networks (N-GENNs) is introduced, a deep learning framework for discovering evolution equations of systems governed by the nonlinear GENERIC formalism, and its ability to accurately infer thermodynamically consistent models from data for systems incorporating both conservative and nonlinear dissipative dynamics is demonstrated.

Vojtěch Votruba, Zequn He, Weilun Qiu et al. · 1 citation

Convergent Evolution: How Different Language Models Learn Similar Number Representations

This paper identifies two different routes through which models can acquire geometrically separable features: they can learn them from complementary co-occurrence signals in general language data, including text-number co-occurrence and cross-number interaction, or from multi-token addition problems.

Deqing Fu, Tianyi Zhou, Mikhail Belkin et al. · 3 citations
#artificial intelligence Preprint Apr 2026

FairNVT: Fair Classification via Noise Injection in Vision Transformers

A lightweight debiasing framework for pretrained transformer-based encoders that improves prediction fairness while preserving task performance and reduces sensitive-attribute attacker accuracy, improves fairness metrics such as demographic parity difference and equalized odds, and maintains competitive task performance.

Qiaoyue Tang, Sepidehsadat Hosseini, Mengyao Zhai et al. · 0 citations

SegWithU: Uncertainty as Perturbation Energy for Single-Forward-Pass Risk-Aware Medical Image Segmentation

SegWithU is a post-hoc framework that augments a frozen pretrained segmentation backbone with a lightweight uncertainty head and models uncertainty as perturbation energy in a compact probe space using rank-1 posterior probes, suggesting that perturbation-based uncertainty modeling is an effective and practical route to reliability-aware medical segmentation.

Tianhao Fu, Austin Wang, Charles D. Chen et al. · 0 citations

Does Unification Come at a Cost? Uni-SafeBench: A Safety Benchmark for Unified Multimodal Large Models

It is found that the original safety alignment of the underlying LLM is not consistently preserved in current unified models, and open-source UMLMs exhibit much lower safety performance than multimodal large models specialized for either generation or understanding tasks, particularly on the generation side.

Zi Peng, Yongxiu Xu, Qinyi Zhang et al. · 0 citations

From tech blogs

See all →
MIT News · Artificial Intelligence Aug 27, 2026

Looking beyond natural sequences

A new machine-learning framework aims to improve the success rate of computational protein design while moving away from results that reproduce sequences found in nature.