Yuan Sui

I am a PhD student at the School of Computing (SoC), National University of Singapore (NUS), majoring in Computer Science. I am fortunate to be supervised by Prof. Byran Hooi. My current research focuses on (1) trustworthiness of large language models (llms); (2) llm reasoning (in specialized fields, graphs, tables, etc.); and (3) causal inference. If you are also passionate about these topics, feel free to connect!

Email  /  Github  /  Google Scholar  /  Twitter  /  LinkedIn

profile photo
Research Interest

A few questions that drive my recent research are:

  • how can we get foundation models to efficiently learn domain knowledge?
  • how can we advance better models with humans' collaborations?
  • how can we reduce potential harms (fairness, privacy and bias)?
  • how can we genuinely advance our understanding of current LLMs (capabilities and limitations), both empirically and theoretically?

News

  • (2025.03): Will present two papers @ ICLR, 2025, BuildingTrust Workshop in Singapore!
  • (2025.02): Invited as a reviewer for NeurIPS'25 & ICML'25!
  • (2025.01): One co-authored paper accepted by WWW'25!
  • (2024.12): Passed Qualifying Exam and became a Ph.D. candidate – thanks to the examiners and my advisor!
  • (2024.12): Earned the Excellent Reviewer award (top 10–20%) at KDD'25!
  • (2024.11): One co-authored paper accepted by KDD'25! Congrats to Yufei!
  • (2024.09): One paper accepted by EMNLP'24!
  • (2024.08): Invited as a reviewer for KDD'24 & ICLR'25!
  • (2024.04): Invited as a reviewer for NeurIPS'24!
  • (2023.10): One paper accepted by WSDM'24! See Microsoft Research Blog of our work!
  • (2023.08): Began my Ph.D. at NUS with a research scholarship!
  • (2023.03): Contributed to Excel Copilot for enhanced data insights!
  • (2022.08): Joined MSRA, DKI Group as a research intern!
  • (2022.06): Interned at Dartmouth's Minds, Machines and Society Lab!
  • (2022.05): One paper accepted by KBS (journal)!
  • (2022.03): One paper accepted by IJCNN'22 (oral)!
  • (2022.02): Interned at ICT’s VIPL Group!

Publications (selected, * refers to equal contribution)
Meta-Reasoner: Dynamic Guidance for Optimized Inference-time Reasoning in Large Language Models
Yuan Sui, Yufei He, Tri Cao, Simeng Han, Bryan Hooi
Preprint, 2025 
arXiv / code
This paper introduce Meta-Reasoner, a framework that dynamically optimizes inference-time reasoning by enabling LLMs to "think about how to think".
UniGraph2: Learning a Unified Embedding Space to Bind Multimodal Graphs
Yufei He, Yuan Sui, Xiaoxin He, Yue Liu, Yifei Sun, Bryan Hooi
WebConf, 2025 (oral)
arXiv / code
This paper introduces UniGraph2, a foundation model that unifies multimodal information and graph structures for effective representation learning on multimodal graphs.
Can Knowledge Graphs Make Large Language Models More Trustworthy? An Empirical Study over Open-ended Question Answering
Yuan Sui, Yufei He, Zifeng Ding, Bryan Hooi
ICLR, BuildingTrust Workshop, 2025 
arXiv / code
This paper introduces OKGQA, a benchmark for evaluating Knowledge Graph-enhanced LLMs in open-ended question answering, focusing on reducing hallucinations and improving reasoning.
FiDeLiS: Faithful Reasoning in Large Language Model for Knowledge Graph Question Answering
Yuan Sui, Yufei He, Nian Liu, Xiaoxin He, Kun Wang, Bryan Hooi
ICLR, BuildingTrust Workshop, 2025 
arXiv / code
This paper presents FiDeLiS, a method combining knowledge graphs and LLMs to enhance reasoning efficiency and reduce hallucinations in downstream QA tasks.
UniGraph: Learning a Unified Cross‑Domain Foundation Model for Text‑Attributed Graphs
Yufei He, Yuan Sui, Xiaoxin He, Bryan Hooi
KDD, 2025 
arXiv / code
This paper introduces UniGraph, a framework for cross-domain graph learning using Text-Attributed Graphs (TAGs) and a proposed pre-training algorithm for effective zero-shot and few-shot learning.
TAP4LLM: Table Provider on Sampling, Augmenting, and Packing Semi-structured Data for Large Language Model Reasoning
Yuan Sui*, Jiaru Zou*, Mengyu Zhou, Xinyi He, Lun Du, Shi Han, Dongmei Zhang
EMNLP, 2024 
arXiv / code / poster / slide
TAP4LLM presents a versatile pre-processor suite for leveraging LLMs in table-based tasks effectively.
Table meets LLM: Can Large Language Models Understand Structured Table Data? A Benchmark and Empirical Study
Yuan Sui, Mengyu Zhou, Mingjie Zhou, Shi Han and Dongmei Zhang
WSDM, 2024
arXiv / code / blog / poster / slide
This paper releases SUC benchmark along with comprehensive empirical studies to evaluate and detect the structural understanding capabilities of LLMs on table data.
Intelligent Predictive Maintenance of Hydraulic Systems based on Virtual Knowledge Graph
Wei Yan, Yu Shi, Zengyan Ji, Yuan Sui, Zhenzhen Tian, Wanjing Wang, Qiushi Cao
Engineering Applications of Artificial Intelligence,  2023 (IF=8)
This paper proposes a virtual knowledge graph for the digital modeling and intelligent predicate maintenance of hydraulic systems in manufacturing.
Causality-aware Enhanced Model for Multi-hop Question Answering over Knowledge Graphs
Yuan Sui, Shanshan Feng, Huaxiang Zhang, Jian Cao, Liang Hu, Nengjun Zhu
Knowledge-Based Systems (KBS), 2022 (IF=8.139)
This paper propose CF-KGQA to improve KGQA tasks by address spurious relations through causal interference.
Trigger-GNN: A Trigger-Based Graph Neural Network for Nested Named Entity Recognition
Yuan Sui, Fanyang Bu, Yingting Hu, Wei Yan, Liang Zhang
IJCNN, 2022 (oral)
Trigger-GNN proposes entity triggers to improve the nested NER task.
Experiences
Data, Knowledge, and Intelligence Group, Microsoft Research Asia (MSRA)
Research Intern | Aug. 2022 to June. 2023
Advisor: Dr. Mengyu Zhou
Minds, Machines and Society Lab, Dartmouth College
Research Intern | June. 2022 to Feb. 2023
Advisor: Prof. Soroush Vosoughi
Visual Information Processing and Learning (VIPL) Lab, ICT, Chinese Academy of Sciences
Research Intern | Feb. 2022 to June. 2022
Advisor: Prof. Shuhui Wang
Academic Service

Teaching

  • 2024/2025 semester 1, Knowledge Discovery and Data Mining (CS54225/CS5425)
  • 2024 Summer, Summer Research Program Mentorship (CP2107, Odyssey)
  • 2025 Summer, Summer Research Program Mentorship (CP2107, Odyssey)

Thanks Jon Barron for this amazing template.