← Search

Kaitao Song

30 accepted papers

2026

Improving Long-Context Summarization with Multi-Granularity Retrieval Optimization

AAAI 2026technical

Retrieval-Augmented Generation (RAG) is an effective solution to overcome the limitations of Large Language Models (LLMs) in terms of specific-domain knowledge and timely information updates. However, current RAG methods typically respond to queries based on isolated segments, lacking the ability to

Cited by 0SourcePDFScholar
2026

Linking Process to Outcome: Conditional Reward Modeling for LLM Reasoning

ICLR 2026poster

Process Reward Models (PRMs) have emerged as a promising approach to enhance the reasoning capabilities of large language models (LLMs) by guiding their step-by-step reasoning toward a final answer. However, existing PRMs either treat each reasoning step in isolation, failing to capture inter-step d…

Cited by 0SourceScholar
2026

Trade in Minutes! Rationality-Driven Agentic System for Quantitative Financial Trading

ICLR 2026poster

Recent advancements in large language models (LLMs) and agentic systems have shown exceptional decision-making capabilities, revealing significant potential for autonomic finance. Current financial trading agents predominantly simulate anthropomorphic roles that inadvertently introduce emotional bia…

Cited by 0SourceScholar
2025

Chain-of-Model Learning for Language Model

NeurIPS 2025poster

In this paper, we propose a novel learning paradigm, termed *Chain-of-Model* (CoM), which incorporates the causal relationship into the hidden states of each layer as a chain style. thereby introducing great scaling efficiency in model training and inference flexibility in deployment.We introduce th…

Cited by 0SourceScholar
2025

EASYTOOL: Enhancing LLM-based Agents with Concise Tool Instruction

NAACL 2025long

There has been a rising interest in utilizing tools in applications of autonomous agents based on large language models (LLMs) to address intricate real-world tasks. To develop LLMbased agents, it usually requires LLMs to understand many tool functions from different tool documentations. However, th…

2025

EvoAgent: Towards Automatic Multi-Agent Generation via Evolutionary Algorithms

NAACL 2025long

The rise of powerful large language models (LLMs) has spurred a new trend in building LLM-based autonomous agents for solving complex tasks, especially multi-agent systems. Despite the remarkable progress, we notice that existing works are heavily dependent on human-designed frameworks, which greatl…

2025

Let LRMs Break Free from Overthinking via Self-Braking Tuning

NeurIPS 2025poster

Large reasoning models (LRMs), such as OpenAI o1 and DeepSeek-R1, have significantly enhanced their reasoning capabilities by generating longer chains of thought, demonstrating outstanding performance across a variety of tasks. However, this performance gain comes at the cost of a substantial increa…

Cited by 0SourceScholar
2025

Mind the Gap: Bridging Thought Leap for Improved Chain-of-Thought Tuning

NeurIPS 2025poster

Large language models (LLMs) have achieved remarkable progress on mathematical tasks through Chain-of-Thought (CoT) reasoning. However, existing mathematical CoT datasets often suffer from **Thought Leaps** due to experts omitting intermediate steps, which negatively impacts model learning and gener…

Cited by 0SourceScholar
2024

Can Graph Learning Improve Planning in LLM-based Agents?

NeurIPS 2024poster

Task planning in language agents is emerging as an important research topic alongside the development of large language models (LLMs). It aims to break down complex user requests in natural language into solvable sub-tasks, thereby fulfilling the original requests. In this context, the sub-tasks can…

2024

Connecting Large Language Models with Evolutionary Algorithms Yields Powerful Prompt Optimizers

ICLR 2024poster

Large Language Models (LLMs) excel in various tasks, but they rely on carefully crafted prompts that often demand substantial human effort. To automate this process, in this paper, we propose a novel framework for discrete prompt optimization, called EvoPrompt, which borrows the idea of evolutionary…

Cited by 0SourcePDFScholar
2024

Improving Large Language Models in Event Relation Logical Prediction

ACL 2024long

Event relations are crucial for narrative understanding and reasoning. Governed by nuanced logic, event relation extraction (ERE) is a challenging task that demands thorough semantic understanding and rigorous logical reasoning. In this paper, we conduct an in-depth investigation to systematically e…

2024

NaturalSpeech 3: Zero-Shot Speech Synthesis with Factorized Codec and Diffusion Models

ICML 2024oral

While recent large-scale text-to-speech (TTS) models have achieved significant progress, they still fall shorts in speech quality, similarity, and prosody. Considering that speech intricately encompasses various attributes (e.g., content, prosody, timbre, and acoustic details) that pose significant…

Cited by 172SourcePDFScholar
2024

PromptTTS 2: Describing and Generating Voices with Text Prompt

ICLR 2024poster

Speech conveys more information than text, as the same word can be uttered in various voices to convey diverse information. Compared to traditional text-to-speech (TTS) methods relying on speech prompts (reference speech) for voice variability, using text prompts (descriptions) is more user-friendly…

2024

TaskBench: Benchmarking Large Language Models for Task Automation

NeurIPS 2024poster

In recent years, the remarkable progress of large language models (LLMs) has sparked interest in task automation, which involves decomposing complex tasks described by user instructions into sub-tasks and invoking external tools to execute them, playing a central role in autonomous agents. However,…

2023

A Hierarchical Regression Chain Framework for Affective Vocal Burst Recognition

ICASSP 2023accepted

As a common way of emotion signaling via non-linguistic vocalizations, vocal burst (VB) plays an important role in daily social interaction. Understanding and modeling human vocal bursts are indispensable for developing robust and general artificial intelligence. Exploring computational approaches f…

Cited by 0SourceScholar
2023

CircuitNet: A Generic Neural Network to Realize Universal Circuit Motif Modeling

ICML 2023poster

The successes of artificial neural networks (ANNs) are largely attributed to mimicking the human brain structures. Recent advances in neuroscience revealed that neurons interact with each other through various kinds of connectivity patterns to process information, in which the common connectivity pa…

Cited by 0SourcePDFScholar
2023

DiffusionNER: Boundary Diffusion for Named Entity Recognition

ACL 2023long

In this paper, we propose DiffusionNER, which formulates the named entity recognition task as a boundary-denoising diffusion process and thus generates named entities from noisy spans. During training, DiffusionNER gradually adds noises to the golden entity boundaries by a fixed forward diffusion pr…

2023

HuggingGPT: Solving AI Tasks with ChatGPT and its Friends in Hugging Face

NeurIPS 2023poster

Solving complicated AI tasks with different domains and modalities is a key step toward artificial general intelligence. While there are numerous AI models available for various domains and modalities, they cannot handle complicated AI tasks autonomously. Considering large language models (LLMs) hav…

2023

Leveraging Pretrained Representations With Task-Related Keywords for Alzheimer's Disease Detection

ICASSP 2023accepted

With the global population aging rapidly, Alzheimer’s disease (AD) is particularly prominent in older adults, which has an insidious onset and leads to a gradual, irreversible deterioration in cognitive domains (memory, communication, etc.). Speech-based AD detection opens up the possibility of wide…

Cited by 0SourceScholar
2023

SoftCorrect: Error Correction with Soft Detection for Automatic Speech Recognition

AAAI 2023technical

Error correction in automatic speech recognition (ASR) aims to correct those incorrect words in sentences generated by ASR models. Since recent ASR models usually have low word error rate (WER), to avoid affecting originally correct tokens, error correction models should only modify incorrect words,…

2023

Towards Understanding Omission in Dialogue Summarization

ACL 2023long

Dialogue summarization aims to condense the lengthy dialogue into a concise summary, and has recently achieved significant progress. However, the result of existing methods is still far from satisfactory. Previous works indicated that omission is a major factor in affecting the quality of summarizat…

2022

A Study on the Efficacy of Model Pre-Training In Developing Neural Text-to-Speech System

ICASSP 2022accepted

In the development of neural text-to-speech systems, model pre-training with a large amount of non-target speakers’ data is a common approach. However, in terms of ultimately achieved system performance for target speaker(s), the actual benefits of model pre-training are uncertain and unstable, depe…

Cited by 0SourceScholar
2022

Analyzing and Mitigating Interference in Neural Architecture Search

ICML 2022spotlight

Weight sharing is a popular approach to reduce the training cost of neural architecture search (NAS) by reusing the weights of shared operators from previously trained child models. However, the rank correlation between the estimated accuracy and ground truth accuracy of those child models is low du…

Cited by 0SourcePDFScholar
2022

Transcormer: Transformer for Sentence Scoring with Sliding Language Modeling

NeurIPS 2022accept

Sentence scoring aims at measuring the likelihood score of a sentence and is widely used in many natural language processing scenarios, like reranking, which is to select the best sentence from multiple candidates. Previous works on sentence scoring mainly adopted either causal language modeling (CL…

Cited by 13SourcePDFScholar
2021

DeepRapper: Neural Rap Generation with Rhyme and Rhythm Modeling

ACL 2021long

Rap generation, which aims to produce lyrics and corresponding singing beats, needs to model both rhymes and rhythms. Previous works for rap generation focused on rhyming lyrics, but ignored rhythmic beats, which are important for rap performance. In this paper, we develop DeepRapper, a Transformer-…

2021

Pyramid Vision Transformer: A Versatile Backbone for Dense Prediction Without Convolutions

ICCV 2021poster

Although convolutional neural networks (CNNs) have achieved great success in computer vision, this work investigates a simpler, convolution-free backbone network useful for many dense prediction tasks. Unlike the recently-proposed Vision Transformer (ViT) that was designed for image classification s…

Cited by 5162PDFcodeScholar
2021

SongMASS: Automatic Song Writing with Pre-training and Alignment Constraint

AAAI 2021technical

Automatic song writing aims to compose a song (lyric and/or melody) by machine, which is an interesting topic in both academia and industry. In automatic song writing, lyric-to-melody generation and melody-to-lyric generation are two important tasks, both of which usually suffer from the following c…

2020

MPNet: Masked and Permuted Pre-training for Language Understanding

NeurIPS 2020poster

BERT adopts masked language modeling (MLM) for pre-training and is one of the most successful pre-training models. Since BERT neglects dependency among predicted tokens, XLNet introduces permuted language modeling (PLM) for pre-training to address this problem. However, XLNet does not leverage the f…

2019

MASS: Masked Sequence to Sequence Pre-training for Language Generation

ICML 2019oral

Pre-training and fine-tuning, e.g., BERT \citep{devlin2018bert}, have achieved great success in language understanding by transferring knowledge from rich-resource pre-training task to the low/zero-resource downstream tasks. Inspired by the success of BERT, we propose MAsked Sequence to Sequence pre…