← Search

Zhaocheng Zhu

13 accepted papers

2026

Landscape of Thoughts: Visualizing the Reasoning Process of Large Language Models

ICLR 2026poster

Numerous applications of large language models (LLMs) rely on their ability to perform step-by-step reasoning. However, the reasoning behavior of LLMs remains poorly understood, posing challenges to research, development, and safety. To address this gap, we introduce landscape of thoughts (LoT), the…

Cited by 0SourcecodeScholar
2025

From Passive to Active Reasoning: Can Large Language Models Ask the Right Questions under Incomplete Information?

ICML 2025poster

While existing benchmarks probe the reasoning abilities of large language models (LLMs) across diverse domains, they predominantly assess passive reasoning, providing models with all the information needed to reach a solution. By contrast, active reasoning—where an LLM must interact with external sy…

2025

Fully-inductive Node Classification on Arbitrary Graphs

ICLR 2025poster

One fundamental challenge in graph machine learning is generalizing to new graphs. Many existing methods following the inductive setup can generalize to test graphs with new structures, but assuming the feature and label spaces remain the same as the training ones. This paper introduces a fully-ind…

2025

Overcoming Long Context Limitations of State Space Models via Context Dependent Sparse Attention

NeurIPS 2025poster

Efficient long-context modeling remains a critical challenge for natural language processing (NLP), as the time complexity of the predominant Transformer architecture scales quadratically with the sequence length. While state-space models (SSMs) offer alternative sub-quadratic solutions, they strugg…

Cited by 0SourcecodeScholar
2024

A Foundation Model for Zero-shot Logical Query Reasoning

NeurIPS 2024poster

Complex logical query answering (CLQA) in knowledge graphs (KGs) goes beyond simple KG completion and aims at answering compositional queries comprised of multiple projections and logical operations. Existing CLQA methods that learn parameters bound to certain entity or relation vocabularies can onl…

2024

Towards Foundation Models for Knowledge Graph Reasoning

ICLR 2024poster

Foundation models in language and vision have the ability to run inference on any textual and visual inputs thanks to the transferable representations such as a vocabulary of tokens in language. Knowledge graphs (KGs) have different entity and relation vocabularies that generally do not overlap. Th…

2024

Towards Foundational Models for Molecular Learning on Large-Scale Multi-Task Datasets

ICLR 2024poster

Recently, pre-trained foundation models have enabled significant advancements in multiple fields. In molecular machine learning, however, where datasets are often hand-curated, and hence typically small, the lack of datasets with labeled features, and codebases to manage those datasets, has hindered…

2023

A*Net: A Scalable Path-based Reasoning Approach for Knowledge Graphs

NeurIPS 2023poster

Reasoning on large-scale knowledge graphs has been long dominated by embedding methods. While path-based methods possess the inductive capacity that embeddings lack, their scalability is limited by the exponential number of paths. Here we present A\*Net, a scalable path-based method for knowledge gr…

2022

Inductive Logical Query Answering in Knowledge Graphs

NeurIPS 2022accept

Formulating and answering logical queries is a standard communication interface for knowledge graphs (KGs). Alleviating the notorious incompleteness of real-world KGs, neural methods achieved impressive results in link prediction and complex query answering tasks by learning representations of enti…

2022

Neural-Symbolic Models for Logical Queries on Knowledge Graphs

ICML 2022spotlight

Answering complex first-order logic (FOL) queries on knowledge graphs is a fundamental task for multi-hop reasoning. Traditional symbolic methods traverse a complete knowledge graph to extract the answers, which provides good interpretation for each step. Recent neural methods learn geometric embedd…

2022

PEER: A Comprehensive and Multi-Task Benchmark for Protein Sequence Understanding

NeurIPS 2022accept

We are now witnessing significant progress of deep learning methods in a variety of tasks (or datasets) of proteins. However, there is a lack of a standard benchmark to evaluate the performance of different methods, which hinders the progress of deep learning in this field. In this paper, we propose…

2021

Neural Bellman-Ford Networks: A General Graph Neural Network Framework for Link Prediction

NeurIPS 2021poster

Link prediction is a very fundamental task on graphs. Inspired by traditional path-based methods, in this paper we propose a general and flexible representation learning framework based on paths for link prediction. Specifically, we define the representation of a pair of nodes as the generalized sum…

2020

GraphAF: a Flow-based Autoregressive Model for Molecular Graph Generation

ICLR 2020poster

Molecular graph generation is a fundamental problem for drug discovery and has been attracting growing attention. The problem is challenging since it requires not only generating chemically valid molecular structures but also optimizing their chemical properties in the meantime. Inspired by the rece…

Cited by 544SourcecodeScholar