Kaiyu Yang

dblp:177/9276 · DBLP profile ↗
← Back
16ranked-venue papers
8as first author
11since 2021 · last 2026
—ORCID · conflict

Domains — the database's venue-derived domains; a paper can count in several

Artificial intelligence and machine learning · 13 · 7 first-author · 8 since 2021Graphics, computer vision, multimedia, augmented reality and games · 3 · 1 first-author · 1 since 2021Systems, architecture and hardware · 1 · 1 first-author · 1 since 2021Software engineering, systems software and programming languages · 1 · 1 since 2021Databases, data management, data science and information retrieval · 1 · 1 since 2021Theory of computation · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
11 papers
Language models and text generation · 24% Knowledge representation and reasoning · 16% Information extraction and text analysis · 14%
Theoretical computer science
6 papers
Automated reasoning and model checking · 86% Computational geometry · 14%
Software engineering, system software, and programming languages
4 papers
Compilers and program optimization · 40% Empirical software engineering · 39% Program analysis · 21%

Topics — the 30 heaviest of 39, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Automated reasoning and model checking
theorem proving
2.742025
Proving Olympiad Inequalities by Synergizing LLMs and Symbolic Reasoning · ICLR 2025
Autoformalizing Euclidean Geometry · ICML 2024
LeanDojo: Theorem Proving with Retrieval-Augmented Language Models · NeurIPS 2023
Knowledge, reasoning and agents › Knowledge representation and reasoning
neuro-symbolic reasoning
0.912025
Proving Olympiad Inequalities by Synergizing LLMs and Symbolic Reasoning · ICLR 2025
Computational geometry
geometric reasoning
0.912025
PyEuclid: A Versatile Formal Plane Geometry System in Python · CAV (4) 2025
Natural language and speech › Language models and text generation
instruction tuning
0.812024
SciInstruct: a Self-Reflective Instruction Annotated Dataset for Training Scientific Language Models · NeurIPS 2024
Natural language and speech › Language models and text generation › pre-trained language model › domain-specific language model
scientific language models
0.812024
SciInstruct: a Self-Reflective Instruction Annotated Dataset for Training Scientific Language Models · NeurIPS 2024
Natural language and speech › Language models and text generation › complex reasoning
scientific reasoning
0.812024
SciInstruct: a Self-Reflective Instruction Annotated Dataset for Training Scientific Language Models · NeurIPS 2024
Automated reasoning and model checking › automated reasoning › mathematical reasoning
autoformalization
0.812024
Autoformalizing Euclidean Geometry · ICML 2024
Automated reasoning and model checking › theorem proving
geometry theorem proving
0.812024
Autoformalizing Euclidean Geometry · ICML 2024
Computer vision › 3D vision › 3d generation
3d scene generation
0.712023
Infinite Photorealistic Worlds Using Procedural Generation · CVPR 2023
Natural language and speech › Language models and text generation
large language model
0.712023
LeanDojo: Theorem Proving with Retrieval-Augmented Language Models · NeurIPS 2023
Knowledge, reasoning and agents › Knowledge representation and reasoning › automated reasoning
premise selection
0.712023
LeanDojo: Theorem Proving with Retrieval-Augmented Language Models · NeurIPS 2023
Natural language and speech › Question answering and dialogue systems
multi-hop reasoning
0.612022
Generating Natural Language Proofs with Verifier-Guided Search · EMNLP 2022
Machine learning › Trustworthy machine learning
privacy and data protection
0.612022
A Study of Face Obfuscation in ImageNet · ICML 2022
Knowledge, reasoning and agents › Knowledge representation and reasoning › automated reasoning
proof search
0.612022
Generating Natural Language Proofs with Verifier-Guided Search · EMNLP 2022
Computer vision › Image recognition and object detection
visual recognition
0.612022
A Study of Face Obfuscation in ImageNet · ICML 2022
Automated reasoning and model checking › theorem proving
proof generation
0.612022
Generating Natural Language Proofs with Verifier-Guided Search · EMNLP 2022
Computer vision › 3D vision
3d scene understanding
0.412020
Rel3D: A Minimally Contrastive Benchmark for Grounding Spatial Relations in 3D · NeurIPS 2020
Natural language and speech › Information extraction and text analysis › syntactic parsing
constituency parsing
0.412020
Strongly Incremental Constituency Parsing with Graph Neural Networks · NeurIPS 2020
Natural language and speech › Information extraction and text analysis › syntactic parsing
incremental parsing
0.412020
Strongly Incremental Constituency Parsing with Graph Neural Networks · NeurIPS 2020
Robotics › Robot navigation and mapping › spatial cognition › spatial knowledge
spatial relation
0.412020
Rel3D: A Minimally Contrastive Benchmark for Grounding Spatial Relations in 3D · NeurIPS 2020
Computer vision › Vision and language › visual grounding
spatial relation grounding
0.412020
Rel3D: A Minimally Contrastive Benchmark for Grounding Spatial Relations in 3D · NeurIPS 2020
Natural language and speech › Information extraction and text analysis
syntactic parsing
0.412020
Strongly Incremental Constituency Parsing with Graph Neural Networks · NeurIPS 2020
Natural language and speech › Information extraction and text analysis › syntactic parsing
transition-based parsing
0.412020
Strongly Incremental Constituency Parsing with Graph Neural Networks · NeurIPS 2020
Computer vision › Vision and language
visual relationship detection
0.412020
Rel3D: A Minimally Contrastive Benchmark for Grounding Spatial Relations in 3D · NeurIPS 2020
Computer vision › Vision and language
spatial reasoning benchmark
0.412019
SpatialSense: An Adversarially Crowdsourced Benchmark for Spatial Relation Recognition · ICCV 2019
Computer vision › 3D vision › 3d scene understanding › spatial relation understanding
spatial relation recognition
0.412019
SpatialSense: An Adversarially Crowdsourced Benchmark for Spatial Relation Recognition · ICCV 2019
Computer vision › Image recognition and object detection
visual relationship recognition
0.412019
SpatialSense: An Adversarially Crowdsourced Benchmark for Spatial Relation Recognition · ICCV 2019
Compilers and program optimization
code generation
0.412019
Learning to Prove Theorems via Interacting with Proof Assistants · ICML 2019
Automated reasoning and model checking › theorem proving
interactive theorem proving
0.412019
Learning to Prove Theorems via Interacting with Proof Assistants · ICML 2019
Machine learning › Deep learning architectures and training › convolutional neural network › convolutional neural network architecture
hourglass network
0.212016
Stacked Hourglass Networks for Human Pose Estimation · ECCV (8) 2016

Methods — techniques the papers use, named apart from their topics

large language model · 2.5retrieval-augmented generation · 2.0hard negative mining · 2.0symbolic reasoning · 1.7proof search · 1.7randomized mathematical rules · 1.3procedural generation · 1.3inference rules · 0.9deductive database · 0.9algebraic solving · 0.9self-reflective instruction annotation · 0.8lean proof assistant · 0.8fine-tuning · 0.8SMT solvers · 0.8stepwise generation · 0.6minimally contrastive data collection · 0.4proof assistant interaction · 0.4deep learning · 0.4
YearPublicationVenuePosition
2026 SGAN: Shapelet-based GAN for local time series black-box attack
Kaiyu Yang, Jidong Yuan, Haoyu Yan, Yongqi Sun
Pattern Recognit.1
2025 PyEuclid: A Versatile Formal Plane Geometry System in Python
abstract
Abstract We introduce , a unified and versatile Python-based formal system for representing and reasoning about plane geometry problems. designs a new formal language that faithfully encodes geometric information, including diagrams, and integrates two complementary components to perform geometric reasoning: (1) a deductive database with an extensive set of inference rules for geometric properties, and (2) an algebraic system for solving diverse equations involving geometric quantities. By seamlessly combining these components, enables human-like reasoning and supports generating concise reasoning steps (proofs), either fully automatically or through interactive guidance. Benchmark evaluations demonstrate that outperforms existing tools, solving a broader range of problems across both proof generation and calculation tasks. Moreover, holds significant potential for educational use and integration with advanced deep learning systems.
Hangrui Bi, Zenan Li, Kaiyu Yang, Xujie Si
CAV (4)5
2025 Proving Olympiad Inequalities by Synergizing LLMs and Symbolic Reasoning
abstract
Large language models (LLMs) can prove mathematical theorems formally by generating proof steps (\textit{a.k.a.} tactics) within a proof system. However, the space of possible tactics is vast and complex, while the available training data for formal proofs is limited, posing a significant challenge to LLM-based tactic generation. To address this, we introduce a neuro-symbolic tactic generator that synergizes the mathematical intuition learned by LLMs with domain-specific insights encoded by symbolic methods. The key aspect of this integration is identifying which parts of mathematical reasoning are best suited to LLMs and which to symbolic methods. While the high-level idea of neuro-symbolic integration is broadly applicable to various mathematical problems, in this paper, we focus specifically on Olympiad inequalities (Figure~1). We analyze how humans solve these problems and distill the techniques into two types of tactics: (1) scaling, handled by symbolic methods, and (2) rewriting, handled by LLMs. In addition, we combine symbolic tools with LLMs to prune and rank the proof goals for efficient proof search. We evaluate our framework on 161 challenging inequalities from multiple mathematics competitions, achieving state-of-the-art performance and significantly outperforming existing LLM and symbolic approaches without requiring additional training data.
Zenan Li, Yuan Yao 0001, Xujie Si, Kaiyu Yang, Xiaoxing Ma
ICLR8
2024 An In-Depth Comparison of Neural and Probabilistic Tree Models for Learning-to-rank
Haonan Tan, Kaiyu Yang, Hai-Tao Yu 0003
ECIR (3)2
2024 Autoformalizing Euclidean Geometry
abstract
Autoformalization involves automatically translating informal math into formal theorems and proofs that are machine-verifiable. Euclidean geometry provides an interesting and controllable domain for studying autoformalization. In this paper, we introduce a neuro-symbolic framework for autoformalizing Euclidean geometry, which combines domain knowledge, SMT solvers, and large language models (LLMs). One challenge in Euclidean geometry is that informal proofs rely on diagrams, leaving gaps in texts that are hard to formalize. To address this issue, we use theorem provers to fill in such diagrammatic information automatically, so that the LLM only needs to autoformalize the explicit textual steps, making it easier for the model. We also provide automatic semantic evaluation for autoformalized theorem statements. We construct LeanEuclid, an autoformalization benchmark consisting of problems from Euclid’s Elements and the UniGeo dataset formalized in the Lean proof assistant. Experiments with GPT-4 and GPT-4V show the capability and limitations of state-of-the-art LLMs on autoformalizing geometry problems. The data and code are available at https://github.com/loganrjmurphy/LeanEuclid.
Logan Murphy, Kaiyu Yang, Anima Anandkumar, Xujie Si
ICML2
2024 Fault Location Method Based on CNN-BiLSTM-Attention for locating Single-phase Ground Fault in Active distribution
abstract
The increasing integration of distributed generation (DG) into the distribution network has led to greater complexity in power flow distribution and transient current characteristics during single-phase grounding faults. As a result, traditional fault location methods are no longer sufficient. Adapting existing fault location methods to accommodate changing permeability has become an urgent issue. In response, a fault location method utilizing a convolutional neural network (CNN) and bidirectional long short-term memory (BiLSTM) network is proposed. The CNN is used to extract detailed longitudinal features from fault zero sequence current data at a specific time, compressing the data length to reduce subsequent network training parameters. Additionally, a cascade network with BiLSTM as its core is constructed to capture historical horizontal features of fault data during the fault evolution process. An Attention mechanism is integrated to ensure that the model focuses on changes in fault time and location data, thereby enhancing fault location accuracy. The simulation results demonstrate that the proposed method is capable of accurately identifying single-phase grounding faults, offering high precision and robustness in positioning, and exhibiting strong adaptability across various permeability fault scenarios.
Kaiyu Yang, Liming Xue, Yirui Sun
IECON1
2024 SciInstruct: a Self-Reflective Instruction Annotated Dataset for Training Scientific Language Models
abstract
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 introduce SciInstruct, a suite of scientific instructions for training scientific language models capable of college-level scientific reasoning. Central to our approach is a novel self-reflective instruction annotation framework to address the data scarcity challenge in the science domain. This framework leverages existing LLMs to generate step-by-step reasoning for unlabelled scientific questions, followed by a process of self-reflective critic-and-revise. Applying this framework, we curated a diverse and high-quality dataset encompassing physics, chemistry, math, and formal proofs. We analyze the curated SciInstruct from multiple interesting perspectives (e.g., domain, scale, source, question type, answer length, etc.). To verify the effectiveness of SciInstruct, we fine-tuned different language models with SciInstruct, i.e., ChatGLM3 (6B and 32B), Llama3-8B-Instruct, and Mistral-7B: MetaMath, enhancing their scientific and mathematical reasoning capabilities, without sacrificing the language understanding capabilities of the base model. We release all codes and SciInstruct at https://github.com/THUDM/SciGLM.
Ziniu Hu, Sining Zhoubian, Zhengxiao Du, Kaiyu Yang, Yisong Yue, Yuxiao Dong, Jie Tang 0001
NeurIPS5
2023 Infinite Photorealistic Worlds Using Procedural Generation
abstract
We introduce Infinigen, a procedural generator of photorealistic 3D scenes of the natural world. Infinigen is entirely procedural: every asset, from shape to texture, is generated from scratch via randomized mathematical rules, using no external source and allowing infinite variation and composition. Infinigen offers broad coverage of objects and scenes in the natural world including plants, animals, terrains, and natural phenomena such as fire, cloud, rain, and snow. Infinigen can be used to generate unlimited, diverse training data for a wide range of computer vision tasks including object detection, semantic segmentation, optical flow, and 3D reconstruction. We expect Infinigen to be a useful resource for computer vision research and beyond. Please visit infinigen.org for videos, code and pre-generated data.
Alexander Raistrick, Lahav Lipson, Zeyu Ma 0004, Lingjie Mei, Yiming Zuo 0001, Karhan Kayan, Hongyu Wen, Beining Han, Alejandro Newell, Hei Law, Ankit Goyal 0001, Kaiyu Yang, Jia Deng 0001
CVPR14
2023 LeanDojo: Theorem Proving with Retrieval-Augmented Language Models
abstract
Large language models (LLMs) have shown promise in proving formal theorems using proof assistants such as Lean. However, existing methods are difficult to reproduce or build on, due to private code, data, and large compute requirements. This has created substantial barriers to research on machine learning methods for theorem proving. This paper removes these barriers by introducing LeanDojo: an open-source Lean playground consisting of toolkits, data, models, and benchmarks. LeanDojo extracts data from Lean and enables interaction with the proof environment programmatically. It contains fine-grained annotations of premises in proofs, providing valuable data for premise selection—a key bottleneck in theorem proving. Using this data, we develop ReProver (Retrieval-Augmented Prover): an LLM-based prover augmented with retrieval for selecting premises from a vast math library. It is inexpensive and needs only one GPU week of training. Our retriever leverages LeanDojo's program analysis capability to identify accessible premises and hard negative examples, which makes retrieval much more effective. Furthermore, we construct a new benchmark consisting of 98,734 theorems and proofs extracted from Lean's math library. It features challenging data split requiring the prover to generalize to theorems relying on novel premises that are never used in training. We use this benchmark for training and evaluation, and experimental results demonstrate the effectiveness of ReProver over non-retrieval baselines and GPT-4. We thus provide the first set of open-source LLM-based theorem provers without any proprietary datasets and release it under a permissive MIT license to facilitate further research.
Kaiyu Yang, Aidan M. Swope, Alex Gu, Rahul Chalamala, Peiyang Song 0002, Shixing Yu, Saad Godil, Ryan Prenger, Anima Anandkumar
NeurIPS1
2022 Generating Natural Language Proofs with Verifier-Guided Search
abstract
Reasoning over natural language is a challenging problem in NLP.In this work, we focus on proof generation: Given a hypothesis and a set of supporting facts, the model generates a proof tree indicating how to derive the hypothesis from supporting facts.Compared to generating the entire proof in one shot, stepwise generation can better exploit the compositionality and generalize to longer proofs but has achieved limited success on real-world data.Existing stepwise methods struggle to generate proof steps that are both logically valid and relevant to the hypothesis.Instead, they tend to hallucinate invalid steps given the hypothesis.In this paper, we present a novel stepwise method, NLProofS (Natural Language Proof Search), which learns to generate relevant steps conditioning on the hypothesis.At the core of our approach, we train an independent verifier to check the validity of the proof steps to prevent hallucination.Instead of generating steps greedily, we search for proofs maximizing a global proof score judged by the verifier.NL-ProofS achieves state-of-the-art performance on EntailmentBank and RuleTaker.Specifically, it improves the correctness of predicted proofs from 27.7% to 33.3% in the distractor setting of EntailmentBank, demonstrating the effectiveness of NLProofS in generating challenging human-authored proofs. 1
Kaiyu Yang, Jia Deng 0001, Danqi Chen 0001
EMNLP1
2022 A Study of Face Obfuscation in ImageNet
abstract
Face obfuscation (blurring, mosaicing, etc.) has been shown to be effective for privacy protection; nevertheless, object recognition research typically assumes access to complete, unobfuscated images. In this paper, we explore the effects of face obfuscation on the popular ImageNet challenge visual recognition benchmark. Most categories in the ImageNet challenge are not people categories; however, many incidental people appear in the images, and their privacy is a concern. We first annotate faces in the dataset. Then we demonstrate that face obfuscation has minimal impact on the accuracy of recognition models. Concretely, we benchmark multiple deep neural networks on obfuscated images and observe that the overall recognition accuracy drops only slightly (<= 1.0%). Further, we experiment with transfer learning to 4 downstream tasks (object recognition, scene recognition, face attribute classification, and object detection) and show that features learned on obfuscated images are equally transferable. Our work demonstrates the feasibility of privacy-aware visual recognition, improves the highly-used ImageNet challenge benchmark, and suggests an important path for future visual datasets. Data and code are available at https://github.com/princetonvisualai/imagenet-face-obfuscation.
Kaiyu Yang, Jacqueline Yau, Li Fei-Fei 0001, Jia Deng 0001, Olga Russakovsky
ICML1
2020 Rel3D: A Minimally Contrastive Benchmark for Grounding Spatial Relations in 3D
abstract
Understanding spatial relations (e.g., laptop on table) in visual input is important for both humans and robots. Existing datasets are insufficient as they lack large-scale, high-quality 3D ground truth information, which is critical for learning spatial relations. In this paper, we fill this gap by constructing Rel3D: the first large-scale, human-annotated dataset for grounding spatial relations in 3D. Rel3D enables quantifying the effectiveness of 3D information in predicting spatial relations on large-scale human data. Moreover, we propose minimally contrastive data collection---a novel crowdsourcing method for reducing dataset bias. The 3D scenes in our dataset come in minimally contrastive pairs: two scenes in a pair are almost identical, but a spatial relation holds in one and fails in the other. We empirically validate that minimally contrastive examples can diagnose issues with current relation detection models as well as lead to sample-efficient training. Code and data are available at https://github.com/princeton-vl/Rel3D.
Ankit Goyal 0001, Kaiyu Yang, Jia Deng 0001
NeurIPS2
2020 Strongly Incremental Constituency Parsing with Graph Neural Networks
abstract
Parsing sentences into syntax trees can benefit downstream applications in NLP. Transition-based parsers build trees by executing actions in a state transition system. They are computationally efficient, and can leverage machine learning to predict actions based on partial trees. However, existing transition-based parsers are predominantly based on the shift-reduce transition system, which does not align with how humans are known to parse sentences. Psycholinguistic research suggests that human parsing is strongly incremental—humans grow a single parse tree by adding exactly one token at each step. In this paper, we propose a novel transition system called attach-juxtapose. It is strongly incremental; it represents a partial sentence using a single tree; each action adds exactly one token into the partial tree. Based on our transition system, we develop a strongly incremental parser. At each step, it encodes the partial tree using a graph neural network and predicts an action. We evaluate our parser on Penn Treebank (PTB) and Chinese Treebank (CTB). On PTB, it outperforms existing parsers trained with only constituency trees; and it performs on par with state-of-the-art parsers that use dependency trees as additional training data. On CTB, our parser establishes a new state of the art. Code is available at https://github.com/princeton-vl/attach-juxtapose-parser.
Kaiyu Yang, Jia Deng 0001
NeurIPS1
2019 SpatialSense: An Adversarially Crowdsourced Benchmark for Spatial Relation Recognition
abstract
Understanding the spatial relations between objects in images is a surprisingly challenging task. A chair may be "behind" a person even if it appears to the left of the person in the image (depending on which way the person is facing). Two students that appear close to each other in the image may not in fact be "next to" each other if there is a third student between them. We introduce SpatialSense, a dataset specializing in spatial relation recognition which captures a broad spectrum of such challenges, allowing for proper benchmarking of computer vision techniques. SpatialSense is constructed through adversarial crowdsourcing, in which human annotators are tasked with finding spatial relations that are difficult to predict using simple cues such as 2D spatial configuration or language priors. Adversarial crowdsourcing significantly reduces dataset bias and samples more interesting relations in the long tail compared to existing datasets. On SpatialSense, state-of-the-art recognition models perform comparably to simple baselines, suggesting that they rely on straightforward cues instead of fully reasoning about this complex task. The SpatialSense benchmark provides a path forward to advancing the spatial reasoning capabilities of computer vision systems. The dataset and code are available at https://github.com/princeton-vl/SpatialSense.
Kaiyu Yang, Olga Russakovsky, Jia Deng 0001
ICCV1
2019 Learning to Prove Theorems via Interacting with Proof Assistants
abstract
Humans prove theorems by relying on substantial high-level reasoning and problem-specific insights. Proof assistants offer a formalism that resembles human mathematical reasoning, representing theorems in higher-order logic and proofs as high-level tactics. However, human experts have to construct proofs manually by entering tactics into the proof assistant. In this paper, we study the problem of using machine learning to automate the interaction with proof assistants. We construct CoqGym, a large-scale dataset and learning environment containing 71K human-written proofs from 123 projects developed with the Coq proof assistant. We develop ASTactic, a deep learning-based model that generates tactics as programs in the form of abstract syntax trees (ASTs). Experiments show that ASTactic trained on CoqGym can generate effective tactics and can be used to prove new theorems not previously provable by automated methods. Code is available at https://github.com/princeton-vl/CoqGym.
Kaiyu Yang, Jia Deng 0001
ICML1
2016 Stacked Hourglass Networks for Human Pose Estimation
Alejandro Newell, Kaiyu Yang, Jia Deng 0001
ECCV (8)2