cs.LGJul 13, 2026

Adapting Evidential Neural Networks to Test-Time Neighbor Fusion Improves Molecular Property Prediction

Authors: Cameron GruichWeichi YaoYixin WangBryan Goldsmith

Organizations: Department of Chemical Engineering, University of Michigan, Ann Arbor, Michigan 48109-2136, USA · Department of Statistics, University of Michigan, Ann Arbor, Michigan 48109-1107 USA

Abstract

A trained molecular property model can be refined at test time by correcting each prediction with the measured labels of the most similar training molecules, a retraining-free procedure we call neighbor fusion; evidential neural networks make it principled by using their aleatoric and epistemic uncertainty to parameterize a Bayesian update. Our main contribution, PG-EVIKAL, learns a property-distance metric to re-rank structurally similar neighbors by their property relevance before fusion, building on EVIKAL (scalar Kalman filter) and GP-EVIKAL (Gaussian process variant handling correlated neighbors). Evaluated on 16 molecular datasets, PG-EVIKAL reduces RMSE relative to the evidential model baseline on 14 of them, with a median reduction of 19.4%, and improves calibration; in sequential-assay scenarios it further incorporates newly measured molecules, refining predictions as they arrive without retraining. This work demonstrates that evidential uncertainty decomposition is not merely a calibration objective but an actionable inference resource that enables test-time refinement of molecular property predictions.

Explore similar work

CardsList