← Search

Yingheng Wang

14 accepted papers

2026

AtomWorld: A Benchmark for Evaluating Spatial Reasoning in Large Language Models on Material Structures

ICML 2026poster

Large language models (LLMs) have shown promising potential in materials science, enabling tasks ranging from knowledge retrieval to property prediction. Existing materials science benchmarks mainly focus on perceptual or knowledge-based tasks, largely ignoring the structure modelling tasks, a core …

Cited by 0SourceScholar
2026

Deep Scientific Reasoning under Physical Constraints: Structure-Aware Spectrum Prediction for Electronic Density of States

ICML 2026poster

Structured scientific spectra encode rich physical information while satisfying hard constraints such as conservation and spectral geometry. We study a canonical example, the electronic density of states (eDOS), whose accurate prediction is central to materials discovery. Prior methods often (i) dec…

Cited by 0SourceScholar
2026

Deep Scientific Reasoning under Physical Constraints: Structure-Aware Spectrum Prediction for Electronic Density of States

ICML 2026poster

Structured scientific spectra encode rich physical information while satisfying hard constraints such as conservation and spectral geometry. We study a canonical example, the electronic density of states (eDOS), whose accurate prediction is central to materials discovery. Prior methods often (i) dec…

Cited by 0SourceScholar
2026

HeuriGym: An Agentic Benchmark for LLM-Crafted Heuristics in Combinatorial Optimization

ICLR 2026poster

While Large Language Models (LLMs) have demonstrated significant advancements in reasoning and agent-based problem-solving, current evaluation methodologies fail to adequately assess their capabilities: existing benchmarks either rely on closed-ended questions prone to saturation and memorization, o…

Cited by 0SourcecodeScholar
2026

MIRNet: Integrating Constrained Graph-Based Reasoning with Pre-training for Diagnostic Medical Imaging

AAAI 2026technical

Automated interpretation of medical images demands robust modeling of complex visual-semantic relationships while addressing annotation scarcity, label imbalance, and clinical plausibility constraints. We introduce MIRNet (Medical Image Reasoner Network), a novel framework that integrates self-super

Cited by 0SourcePDFScholar
2026

Machine Learning Methods for Studying Latent Neural Activity Dynamics

IJCAI 2026

Recent developments in brain recording are driving a demand for machine learning tools capable of decoding the latent structure of large populations of neurons. In this paper, we provide a comprehensive survey that outlines the trajectory of Latent Variable Models (LVMs) from early state-space model

Cited by 0Scholar
2026

STELLAR: Spatio-Temporal Environmental Learning with Latent Alignment and Refinement for Long-Tailed Species Distribution Modeling

IJCAI 2026

Joint Species Distribution Modeling (JSDM) is a key enabler for biodiversity monitoring and conservation planning. However, accurate JSDM faces two coupled challenges: environmental drivers and species distributions are inherently spatio-temporal, while species co-occurrence patterns exhibit complex

Cited by 0Scholar
2026

Unsupervised Combinatorial Probabilistic Reasoning: Probabilistic Coin Change Problem

AAAI 2026technical

We introduce the Probabilistic Coin Change Problem (PCCP), a novel variant of the classical Combination Coin Change Problem (CCCP), motivated by a real-world scientific inverse task. The goal of CCCP is to enumerate all unordered combinations of coin denominations that sum to a given target. In PCCP

Cited by 0SourcePDFScholar
2025

Denoising Diffusion Variational Inference: Diffusion Models as Expressive Variational Posteriors

AAAI 2025technical

We propose denoising diffusion variational inference (DDVI), a black-box variational inference algorithm for latent variable models which relies on diffusion models as flexible approximate posteriors. Specifically, our method introduces an expressive class of diffusion-based variational posteriors t…

2024

Conformal Crystal Graph Transformer with Robust Encoding of Periodic Invariance

AAAI 2024technical

Machine learning techniques, especially in the realm of materials design, hold immense promise in predicting the properties of crystal materials and aiding in the discovery of novel crystals with desirable traits. However, crystals possess unique geometric constraints—namely, E(3) invariance for pri…

Cited by 4SourcePDFScholar
2023

InfoDiffusion: Representation Learning Using Information Maximizing Diffusion Models

ICML 2023poster

While diffusion models excel at generating high-quality samples, their latent variables typically lack semantic meaning and are not suitable for representation learning. Here, we propose InfoDiffusion, an algorithm that augments diffusion models with low-dimensional latent variables that capture hig…

Cited by 41SourcePDFScholar
2023

M$^2$Hub: Unlocking the Potential of Machine Learning for Materials Discovery

NeurIPS 2023poster

We introduce M$^2$Hub, a toolkit for advancing machine learning in materials discovery. Machine learning has achieved remarkable progress in modeling molecular structures, especially biomolecules for drug discovery. However, the development of machine learning approaches for modeling materials struc…

2023

Time Series Contrastive Learning with Information-Aware Augmentations

AAAI 2023technical

Various contrastive learning approaches have been proposed in recent years and achieve significant empirical success. While effective and prevalent, contrastive learning has been less explored for time series data. A key component of contrastive learning is to select appropriate augmentations imposi…

2022

Going Deeper into Permutation-Sensitive Graph Neural Networks

ICML 2022spotlight

The invariance to permutations of the adjacency matrix, i.e., graph isomorphism, is an overarching requirement for Graph Neural Networks (GNNs). Conventionally, this prerequisite can be satisfied by the invariant operations over node permutations when aggregating messages. However, such an invariant…