← Search

Ziniu Hu

29 accepted papers

2025

Automated Molecular Concept Generation and Labeling with Large Language Models

COLING 2025main

Artificial intelligence (AI) is transforming scientific research, with explainable AI methods like concept-based models (CMs) showing promise for new discoveries. However, in molecular science, CMs are less common than black-box models like Graph Neural Networks (GNNs), due to their need for predefi…

2025

Optimized Multi-Token Joint Decoding With Auxiliary Model for LLM Inference

ICLR 2025poster

Large language models (LLMs) have achieved remarkable success across diverse tasks, yet their inference processes are hindered by substantial time and energy demands due to single-token generation at each decoding step. While previous methods such as speculative decoding mitigate these inefficiencie…

Cited by 3SourcePDFScholar
2025

QLASS: Boosting Language Agent Inference via Q-Guided Stepwise Search

ICML 2025poster

Language agents have become a promising solution to complex interactive tasks. One of the key ingredients to the success of language agents is the reward model on the trajectory of the agentic workflow, which provides valuable guidance during training or inference. However, due to the lack of annota…

2025

SFS: Smarter Code Space Search improves LLM Inference Scaling

ICLR 2025poster

We frame code generation as a black-box optimization problem within the code space and demonstrate how optimization-inspired techniques can enhance inference scaling over text. Based on this perspective, we propose **SCATTERED FOREST SEARCH (SFS)**, a novel approach that improves solution diversity…

Cited by 0SourcePDFScholar
2025

Self-Evolving Visual Concept Library using Vision-Language Critics

CVPR 2025poster

We study the problem of building a visual concept library for visual recognition. Building effective visual concept libraries is challenging, as manual definition is labor-intensive, while relying solely on LLMs for concept generation can result in concepts that lack discriminative power or fail to…

Cited by 0SourcePDFScholar
2025

Strategist: Self-improvement of LLM Decision Making via Bi-Level Tree Search

ICLR 2025poster

Traditional reinforcement learning and planning require a lot of data and training to develop effective strategies. On the other hand, large language models (LLMs) can generalize well and perform tasks without prior training but struggle with complex planning and decision-making. We introduce **STRA…

Cited by 0SourcePDFScholar
2025

TreeRL: LLM Reinforcement Learning with On-Policy Tree Search

ACL 2025long

Reinforcement learning (RL) with tree search has demonstrated superior performance in traditional reasoning tasks. Compared to conventional independent chain sampling strategies with outcome supervision, tree search enables better exploration of the reasoning space and provides dense, on-policy proc…

2024

Can Large Language Model Agents Simulate Human Trust Behavior?

NeurIPS 2024poster

Large Language Model (LLM) agents have been increasingly adopted as simulation tools to model humans in social science and role-playing applications. However, one fundamental question remains: can LLM agents really simulate human behavior? In this paper, we focus on one critical and elemental behavi…

2024

Enhancing Large Vision Language Models with Self-Training on Image Comprehension

NeurIPS 2024poster

Large vision language models (LVLMs) integrate large language models (LLMs) with pre-trained vision encoders, thereby activating the perception capability of the model to understand image inputs for different queries and conduct subsequent reasoning. Improving this capability requires high-quality v…

2024

Physics-Informed Regularization for Domain-Agnostic Dynamical System Modeling

NeurIPS 2024poster

Learning complex physical dynamics purely from data is challenging due to the intrinsic properties of systems to be satisfied. Incorporating physics-informed priors, such as in Hamiltonian Neural Networks (HNNs), achieves high-precision modeling for energy-conservative systems. However, real-world s…

2024

ReST-MCTS*: LLM Self-Training via Process Reward Guided Tree Search

NeurIPS 2024poster

Recent methodologies in LLM self-training mostly rely on LLM generating responses and filtering those with correct output answers as training data. This approach often yields a low-quality fine-tuning training set (e.g., incorrect plans or intermediate reasoning). In this paper, we develop a reinfor…

2024

SceneCraft: An LLM Agent for Synthesizing 3D Scenes as Blender Code

ICML 2024oral

This paper introduces SceneCraft, a Large Language Model (LLM) Agent converting text descriptions into Blender-executable Python scripts which render complex scenes with up to a hundred 3D assets. This process requires complex spatial planning and arrangement. We tackle these challenges through a co…

Cited by 39SourcePDFScholar
2024

SciBench: Evaluating College-Level Scientific Problem-Solving Abilities of Large Language Models

ICML 2024poster

Most existing Large Language Model (LLM) benchmarks on scientific problem reasoning focus on problems grounded in high-school subjects and are confined to elementary algebraic operations. To systematically examine the reasoning capabilities required for solving complex scientific problems, we introd…

2024

SciInstruct: a Self-Reflective Instruction Annotated Dataset for Training Scientific Language Models

NeurIPS 2024poster

Large Language Models (LLMs) have shown promise in assisting scientific discovery. However, such applications are currently limited by LLMs' deficiencies in understanding intricate scientific concepts, deriving symbolic equations, and solving advanced numerical calculations. To bridge these gaps, we…

2024

Symbolic Music Generation with Non-Differentiable Rule Guided Diffusion

ICML 2024oral

We study the problem of symbolic music generation (e.g., generating piano rolls), with a technical focus on non-differentiable rule guidance. Musical rules are often expressed in symbolic form on note characteristics, such as note density or chord progression, many of which are non-differentiable wh…

2023

AVIS: Autonomous Visual Information Seeking with Large Language Model Agent

NeurIPS 2023poster

In this paper, we propose an autonomous information seeking visual question answering framework, AVIS. Our method leverages a Large Language Model (LLM) to dynamically strategize the utilization of external tools and to investigate their outputs via tree search, thereby acquiring the indispensable k…

Cited by 51SourcePDFScholar
2023

REVEAL: Retrieval-Augmented Visual-Language Pre-Training With Multi-Source Multimodal Knowledge Memory

CVPR 2023highlight

In this paper, we propose an end-to-end Retrieval-Augmented Visual Language Model (REVEAL) that learns to encode world knowledge into a large-scale memory, and to retrieve from it to answer knowledge-intensive queries. REVEAL consists of four key components: the memory, the encoder, the retriever an…

2023

Towards a Comprehensive Benchmark for High-Level Synthesis Targeted to FPGAs

NeurIPS 2023poster

High-level synthesis (HLS) aims to raise the abstraction layer in hardware design, enabling the design of domain-specific accelerators (DSAs) like field-programmable gate arrays (FPGAs) using C/C++ instead of hardware description languages (HDLs). Compiler directives in the form of pragmas play a cr…

2022

Empowering Language Models with Knowledge Graph Reasoning for Open-Domain Question Answering

EMNLP 2022main

Answering open-domain questions requires world knowledge about in-context entities. As pre-trained Language Models (LMs) lack the power to store all required knowledge, external knowledge sources, such as knowledge graphs, are often used to augment LMs. In this work, we propose knOwledge REasOning e…

Cited by 61SourcePDFScholar
2022

Improving Multi-Task Generalization via Regularizing Spurious Correlation

NeurIPS 2022accept

Multi-Task Learning (MTL) is a powerful learning paradigm to improve generalization performance via knowledge sharing. However, existing studies find that MTL could sometimes hurt generalization, especially when two tasks are less correlated. One possible reason that hurts generalization is spurious…

Cited by 33SourcePDFScholar
2022

Zero-shot Transfer Learning within a Heterogeneous Graph via Knowledge Transfer Networks

NeurIPS 2022accept

Data continuously emitted from industrial ecosystems such as social or e-commerce platforms are commonly represented as heterogeneous graphs (HG) composed of multiple node/edge types. State-of-the-art graph learning methods for HGs known as heterogeneous graph neural networks (HGNNs) are applied to…

2021

Broaden the Vision: Geo-Diverse Visual Commonsense Reasoning

EMNLP 2021main

Commonsense is defined as the knowledge on which everyone agrees. However, certain types of commonsense knowledge are correlated with culture and geographic locations and they are only shared locally. For example, the scenes of wedding ceremonies vary across regions due to different customs influenc…

2020

Emoji-Powered Representation Learning for Cross-Lingual Sentiment Classification (Extended Abstract)

IJCAI 2020poster

Sentiment classification typically relies on a large amount of labeled data. In practice, the availability of labels is highly imbalanced among different languages. To tackle this problem, cross-lingual sentiment classification approaches aim to transfer knowledge learned from one language that has…

2020

Improving Neural Language Generation with Spectrum Control

ICLR 2020poster

Recent Transformer-based models such as Transformer-XL and BERT have achieved huge success on various natural language processing tasks. However, contextualized embeddings at the output layer of these powerful models tend to degenerate and occupy an anisotropic cone in the vector space, which is cal…

Cited by 96SourceScholar
2019

Layer-Dependent Importance Sampling for Training Deep and Large Graph Convolutional Networks

NeurIPS 2019poster

Graph convolutional networks (GCNs) have recently received wide attentions, due to their successful applications in different graph tasks and different domains. Training GCNs for a large graph, however, is still a challenge. Original full-batch GCN training requires calculating the representation of…