← Search

Weijie Ma

2 accepted papers

2026

Optimizing Inference-Time Compute for Medical Reasoning via Uncertainty Quantification

ICML 2026poster

Extended Chain-of-Thought (CoT) reasoning has significantly bolstered the capabilities of medical large language models (LLMs). However, current models exhibit static computational expenditure, applying lengthy reasoning processes indiscriminately to both simple queries and complex diagnostic cases.…

Cited by 0SourceScholar
2026

Periodic Bayesian Flow Networks with Additive Accuracy

ICML 2026poster

Generating periodic data---such as fractional atomic coordinates in crystal structures and phase patterns in compressive light-field (CLF) displays---is challenging because wrap-around boundaries complicate probabilistic modeling and learning. While Bayesian Flow Networks (BFNs) offer a powerful gen…

Cited by 0SourceScholar