← Search

Yujun Yan

16 accepted papers

2026

HalluGuard: Demystifying Data-Driven and Reasoning-Driven Hallucinations in LLMs

ICLR 2026poster

The reliability of Large Language Models (LLMs) in high-stakes domains such as healthcare, law, and scientific discovery is often compromised by hallucinations. These failures typically stem from two sources: *data-driven hallucinations* and *reasoning-driven hallucinations*. However, existing detec…

Cited by 0SourcecodeScholar
2026

RL4RLA: Teaching ML to Discover Randomized Linear Algebra Algorithms through Curriculum Design and Graph-based Search

ICML 2026poster

Randomized linear algebra (RLA) algorithms are essential for scaling scientific computing and machine learning, yet their discovery remains mostly a manual process that requires deep expert knowledge and inspiration. While Reinforcement Learning (RL) offers a pathway to automation, standard approach…

Cited by 0SourceScholar
2026

Seeing Through the Brain: New Insights from Decoding Visual Stimuli with fMRI

ICLR 2026oral

Understanding how the brain encodes visual information is a central challenge in neuroscience and machine learning. A promising approach is to reconstruct visual stimuli—essentially images—from functional Magnetic Resonance Imaging (fMRI) signals. This involves two stages: transforming fMRI signals…

Cited by 0SourceScholar
2026

Unveiling Multi-regime Patterns in SciML: Distinct Failure Modes and Regime-specific Optimization

ICML 2026poster

Neural networks (NNs) trained under different hyperparameters can fall into distinct training ``regimes'', with models in the same regime showing homogeneous properties and models across regimes differing qualitatively. In this paper, we analyze multi-regime patterns in scientific machine learning (…

Cited by 0SourceScholar
2025

GENUINE: Graph Enhanced Multi-level Uncertainty Estimation for Large Language Models

EMNLP 2025

Uncertainty estimation is essential for enhancing the reliability of Large Language Models (LLMs), particularly in high-stakes applications. Existing methods often overlook semantic dependencies, relying on token-level probability measures that fail to capture structural relationships within the gen

2025

Judging with Many Minds: Do More Perspectives Mean Less Prejudice? On Bias Amplification and Resistance in Multi-Agent Based LLM-as-Judge

EMNLP 2025

LLM-as-Judge has emerged as a scalable alternative to human evaluation, enabling large language models (LLMs) to provide reward signals in trainings. While recent work has explored multi-agent extensions such as multi-agent debate and meta-judging to enhance evaluation quality, the question of how i

Cited by 0SourcePDFScholar
2025

MindLLM: A Subject-Agnostic and Versatile Model for fMRI-to-text Decoding

ICML 2025poster

Decoding functional magnetic resonance imaging (fMRI) signals into text has been a key challenge in the neuroscience community, with the potential to advance brain-computer interfaces and uncover deeper insights into brain mechanisms. However, existing approaches often struggle with suboptimal predi…

Cited by 0SourcePDFScholar
2025

Spectral Insights into Data-Oblivious Critical Layers in Large Language Models

ACL 2025finding

Understanding how feature representations evolve across layers in large language models (LLMs) is key to improving their interpretability and robustness. While recent studies have identified critical layers linked to specific functions or behaviors, these efforts typically rely on data-dependent ana…

Cited by 0SourcePDFScholar
2025

Transfer Faster, Price Smarter: Minimax Dynamic Pricing under Cross-Market Preference Shift

NeurIPS 2025spotlight

We study contextual dynamic pricing when a target market can leverage $K$ auxiliary markets—offline logs or concurrent streams—whose *mean utilities differ by a structured preference shift*. We propose *Cross-Market Transfer Dynamic Pricing (CM-TDP)*, the first algorithm that *provably* handles such…

Cited by 0SourceScholar
2024

Enhancing Size Generalization in Graph Neural Networks through Disentangled Representation Learning

ICML 2024poster

Although most graph neural networks (GNNs) can operate on graphs of any size, their classification performance often declines on graphs larger than those encountered during training. Existing methods insufficiently address the removal of size information from graph representations, resulting in sub-…

2024

EvoluNet: Advancing Dynamic Non-IID Transfer Learning on Graphs

ICML 2024poster

Non-IID transfer learning on graphs is crucial in many high-stakes domains. The majority of existing works assume stationary distribution for both source and target domains. However, real-world graphs are intrinsically dynamic, presenting challenges in terms of domain evolution and dynamic discrepan…

2024

Exploring Consistency in Graph Representations: from Graph Kernels to Graph Neural Networks

NeurIPS 2024poster

Graph Neural Networks (GNNs) have emerged as a dominant approach in graph representation learning, yet they often struggle to capture consistent similarity relationships among graphs. To capture similarity relationships, while graph kernel methods like the Weisfeiler-Lehman subtree (WL-subtree) and…

2024

Medformer: A Multi-Granularity Patching Transformer for Medical Time-Series Classification

NeurIPS 2024poster

Medical time series (MedTS) data, such as Electroencephalography (EEG) and Electrocardiography (ECG), play a crucial role in healthcare, such as diagnosing brain and heart diseases. Existing methods for MedTS classification primarily rely on handcrafted biomarkers extraction and CNN-based models, wi…

2024

Sharpness-diversity tradeoff: improving flat ensembles with SharpBalance

NeurIPS 2024poster

Recent studies on deep ensembles have identified the sharpness of the local minima of individual learners and the diversity of the ensemble members as key factors in improving test-time performance. Building on this, our study investigates the interplay between sharpness and diversity within deep en…

Cited by 1SourcePDFScholar
2020

Beyond Homophily in Graph Neural Networks: Current Limitations and Effective Designs

NeurIPS 2020poster

We investigate the representation power of graph neural networks in the semi-supervised node classification task under heterophily or low homophily, i.e., in networks where connected nodes may have different class labels and dissimilar features. Many popular GNNs fail to generalize to this setting,…

2020

Neural Execution Engines: Learning to Execute Subroutines

NeurIPS 2020poster

A significant effort has been made to train neural networks that replicate algorithmic reasoning, but they often fail to learn the abstract concepts underlying these algorithms. This is evidenced by their inability to generalize to data distributions that are outside of their restricted training set…