Skip to content

Category

machine learning

2,891 papers

#machine learning Preprint Aug 2026

PhyMamba: Physics-Modulated Mamba for Robust Battery Health Prognostics

Experiments show that PhyMamba achieves the best aggregated performance, with an overall mean error reduction of 31.8% compared with a diverse range of baselines, which supports practical deployment for robust battery health prognostics.

S. Sameer, Yunyi Zhao, Wei Zhang et al. · 0 citations
#machine learning Preprint Aug 2026

Temporal Memory-Aware Online Test-Time Adaptation on Dynamic Graphs

A novel framework of temporal memory-aware Online Test-Time Adaptation on Dynamic Graphs, named DGOTTA, to effectively adapt well-trained DGNNs during test time and significantly improves generalization under diverse distribution shifts and multiple model architectures is proposed.

Bo Li, Xin Zheng, Ming Jin et al. · 0 citations
#machine learning Preprint Open access Aug 2026

There and Back Again: Bidirectional Diffusion Bridges for Multimodality Translation

Multimodality translation (e.g., text-to-image) is a core generative AI task. However, existing approaches (1) follow generative paths that do not directly represent the source modality, limiting the flexibility of some sampling algorithms; and (2) are unidirectional, preventing inversion (e.g., image-to-text). We propose BIT: Bidirectional Image-Text Diffusion Bridges. In contrast to previous approaches, BIT starts directly from text and interpolates into images, providing (1) a source-aware generative path that enables diverse and flexible sampling algorithms; and (2) an endpoint-conditioned process that can be traversed from image to text, providing a unified, bidirectional generative framework. BIT is derived through stochastic calculus, yielding SDE forms amenable to simulation and tractable loss functions that scale to high dimensions. Our experiments show that BIT is competitive with denoising-diffusion and deterministic-flow baselines, and outperforms them on several vision--language and natural-science evaluations.

Gabe Guo, Elon Litman, Thanawat Sornwanee et al. · 0 citations
#machine learning Preprint Aug 2026

Beyond Pairwise Graphs in Science: Hypergraph Adaptive Wavelet Operators for Parametric PDEs

The Hypergraph Adaptive waveLet Operator (HALO), which lifts the domain to a hypergraph and learns in its spectral wavelet domain, achieves best or near-best accuracy among frequency-, transformer-, DeepONet-, state-space-, and graph-based baselines and sustains stable multi-step rollouts.

R. Sarkar, Venkataramana Runkana, Souvik Chakraborty · 0 citations
#machine learning Preprint Aug 2026

Node-wise Feature Encoding for Neural Performance Prediction

This work introduces FeatureFormer, a neural performance predictor that incorporates explicit node-wise encodings of FLOPs, parameter counts, and memory proxies within a gated graph attention architecture and presents NNEQ, a new large-scale energy consumption dataset that enables unified evaluation of latency and energy prediction.

Matthew Grenier, William Hammer, Andrew Heuer et al. · 0 citations
#machine learning Preprint Aug 2026

Initialization Is Critical: Advancing Federated Short-Term Load Forecasting under Load Heterogeneity via Model Initialization

This paper studies the role of model initialization in federated STLF, and proposes two initialization strategies from global and local perspectives, which effectively improve forecasting performance, as evidenced by reduced client drift, improved convergence behavior, and lower forecasting errors.

Jia-Ning Chen, Vajiheh Farhadi, Yan Li et al. · 0 citations
#machine learning Preprint Aug 2026

Fast Weight Attention for Continual Learning

This framework separates temporal alignment, plasticity, forgetting, and bounded rehearsal in recurrent sequence models, together with numerically stable positive-decay renormalization, to remain competitive in language modeling and improve length extrapolation on variable-digit addition.

Yi-Fan Zhang, Steve Ta, Jasper Zhang et al. · 0 citations
#machine learning Preprint Aug 2026

The Calls are Coming from Inside the Model: Investigating Probe-based Detection of Tool-Calling Errors in LLMs

Overall, it is found that probing is an effective means to catch a range of different tool-calling errors, including errors arising from using an argument that has the wrong value but the correct type, which might not be recorded by standard logging frameworks.

Eric C. Yeats, Brendan Kennedy, Loc Truong et al. · 0 citations
#machine learning Preprint Aug 2026

Diffusion Distillation for Efficient Weather Ensembles

A supervised energy-distance distillation method is introduced that compresses a multi-step diffusion teacher into a single-step student by aligning student forecasts with teacher samples and ground-truth observations and preserves skill for extreme events.

Yiming Yang, Valentin Brekke, James Briant 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.

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