Skip to content

Category

machine learning

3,367 papers

#machine learning Preprint Aug 2026

Margin-Regularized Structured Semantic Alignment for Brain-Language Correspondence

This work proposes a margin-regularized structured semantic alignment framework that directly aligns brain embeddings with text embeddings in a shared semantic space, enabling retrieval-based decoding and enables explicit modeling of the correspondence between neural representations and language semantics.

Jiaqi Wang, Huawen Hu, Shu Zhang · 0 citations
#artificial intelligence Preprint Aug 2026

FedPref: Federated Preference Learning for Structured Radiology Report Extraction

FedPref lets institutions with unequal, unpooled data benefit from collaboration without ever sharing reports or annotations, and frozen public language models propose alternative JSON extractions, local annotations rank them, and sites collaboratively train compact Qwen3-8B adapters while sharing only model updates.

Flint Xiaofeng Fan, Cheston Tan, Y. Ong et al. · 0 citations
#artificial intelligence Preprint Aug 2026

Probing the Prefill: Detecting Code Vulnerabilities via Latent Activations

LLM-based code generation is now embedded in mission-critical pipelines, but defenses against vulnerable output remain post-hoc -- static analyzers, fine-tuned classifiers, or an LLM judge that screen completed code, ignoring the generating model's own internal state. We test a narrower, directly measurable question: when an LLM reads a piece of C/C++ code as context, do its hidden activations already carry a signal about that code's vulnerability status? We extract last prefill token activations from four LLMs (Granite-4.1-8B, Qwen3.5-9B, Qwen3.6-27B, Gemma-4-12B) across three model families and train MLP probes on these activations. We evaluate them on four function-level C/C++ benchmarks (Devign, Big-Vul, Draper VDISC, PrimeVul). Our probes achieve 41.7\% average F1 using 13.4--16.0M-parameter probes -- under 0.2\% of base-model size. On Devign, the best probe (Qwen3.5-9B, 68.8\% F1) matches the published fine-tuned-classifier SOTA (67.9\%) despite reading only a frozen, general-purpose LLM's activations; on the harder, more imbalanced benchmarks (Big-Vul, Draper VDISC, PrimeVul) probes trail SOTA substantially. This is early evidence that a coding LLM's own representation of arbitrary code is informative about that code's vulnerability status, motivating further work toward lightweight, model-native vulnerability screening.

Alizishaan Khatri · 0 citations
#machine learning Preprint Aug 2026

Diagonal Multi-omics Integration of Heterogenous Datasets

A novel characteristic of dataset heterogeneity is introduced by employing the norm of the difference between the maximum and minimum points in the classical terms of functional analysis.

Maksim V.Kukushkin, M. Arbatskiy, D. Balandin et al. · 0 citations
#machine learning Preprint Aug 2026

MagViT: Interpretable Multi-Magnification Transformers with Patient-Level Model Selection for Breast Histopathology

MagViT, an interpretable multi-magnification transformer framework with scale-gated fusion and patient-level model selection, is presented, an interpretable multi-magnification transformer framework with scale-gated fusion and patient-level model selection relative to prior ViT-centered BreakHis work.

Nabil Ashab, Soumitra Kundu, Saif Mahmud Parvez et al. · 0 citations
#machine learning Preprint Aug 2026

WONDER: A Radio World Model-based Negotiation Framework for Multi-Agent UAV Coverage Optimization

A raido World-model-based Optimized Negotiation framework for Distributed UAV covERage (WONDER), which uses a Joint-Embedding Predictive Architecture (JEPA)-based radio world model to learn and predict the incremental radio effect of each candidate trajectory from deployment-available information and builds RadioDynamics, a comprehensive simulation environment that integrates UAV mobility, radio propagation, inter-UAV communication modeling, and digital-twin geometry.

Jiahao Huang, Rongpeng Li, Zhifeng Zhao et al. · 0 citations
#machine learning Preprint Aug 2026

SPSA Hyperparameter Tuning for Variational Quantum Natural Language Inference

Training variational quantum models requires choosing between parameter-shift gradients, which are exact but cost $O(P)$ forward evaluations, and simultaneous perturbation stochastic approximation (SPSA), which uses only two samples but produces high-variance estimates that can degrade optimisation on small supervised tasks. Whether the cheap gradient is usable depends on the variance that results from different choices of the SPSA perturbation scale, learning rate, and gain-decay schedule. We varied those quantities across a broad grid on a 6-qubit, 60-parameter QNLI classifier and compared the best configurations to parameter-shift AdamW and BuresQNG. AdamW-style SPSA with $c_0=0.01$, $\eta=0.10$, $\gamma=0.10$ reached $55\% \pm 11\%$ test accuracy, improving over the default configuration ($49\% \pm 6\%$) but remaining 16-19 percentage points below the parameter-shift baselines because the two-sample SPSA gradient estimate has too much variance for reliable optimisation of 60 parameters in 40 epochs. Classical-gain SPSA and Bures-preconditioned SPSA performed worse, at $51\%$ and $46\%$ respectively. Bures-preconditioning a noisy two-sample SPSA gradient amplifies perturbation noise.

Nayan D'Souza, Christopher J. Agostino · 0 citations
#machine learning Preprint Aug 2026

Network Denoising Revisited: A Ricci-Flow-Inspired Graph Diffusion Method

This work proposes Ricci-Diffusion, a curvature-guided graph diffusion method inspired by Ricci flow, which exhibits a Ricci-flow-like evolution, in which relative edge-level curvature modulates local transport in the diffusion kernel and guides edge-weight updates toward a more regular graph geometry.

Ye Fang, Chuan-Xian Ren · 0 citations
#machine learning Preprint Aug 2026

MITRE-SAGE: A Multi-Agent Cybersecurity Question-Answering Model

This work proposes MITRE-SAGE, a multi-agent retrieval-augmented generation framework that integrates semantic and structural cybersecurity knowledge to improve the reliability and interpretability of LLM-based QA systems and proposes MITRE-QA, a comprehensive benchmark for evaluating LLMs across diverse cybersecurity knowledge tasks.

Ali Habibzadeh, Farid Feyzi, Reza Ebrahimi Atani · 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.

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.