← Search

Song Jiang

11 accepted papers

2026

SPG: Sandwiched Policy Gradient for Masked Diffusion Language Models

ICLR 2026poster

Diffusion large language models (dLLMs) are emerging as an efficient alternative to autoregressive models due to their ability to decode multiple tokens in parallel. However, aligning dLLMs with human preferences or task-specific rewards via reinforcement learning (RL) is challenging because their i…

Cited by 0SourcecodeScholar
2025

100-LongBench: Are de facto Long-Context Benchmarks Literally Evaluating Long-Context Ability?

ACL 2025finding

Long-context capability is considered one of the most important abilities of LLMs, as a truly long context-capable LLM shall enable its users to effortlessly process many originally exhausting tasks — e.g., digesting a long-form document to find answers v.s., directly asking an LLM about it. However…

2025

NaturalReasoning: Reasoning in the Wild with 2.8M Challenging Questions

NeurIPS 2025poster

Scaling reasoning capabilities beyond traditional domains such as math and coding is hindered by the lack of diverse and high-quality questions. To overcome this limitation, we introduce a scalable approach for generating diverse and challenging reasoning questions, accompanied by reference answers.…

Cited by 0SourceScholar
2025

SmartCache: Context-aware Semantic Cache for Efficient Multi-turn LLM Inference

NeurIPS 2025poster

Large Language Models (LLMs) for multi-turn conversations suffer from inefficiency: semantically similar queries across different user sessions trigger redundant computation and duplicate memory-intensive Key-Value (KV) caches. Existing optimizations such as prefix caching overlook semantic similari…

Cited by 0SourceScholar
2024

A Single Vector Is Not Enough: Taxonomy Expansion via Box Embeddings (Extended Abstract)

IJCAI 2024poster

Taxonomies support various practical web applications such as product navigation in online shopping and user profile tagging on social platforms. Most existing methods for expanding taxonomies encode entities into vector embeddings (i.e., single points). However, we argue that vectors are insufficie…

Cited by 0SourcePDFScholar
2024

FUSE: Measure-Theoretic Compact Fuzzy Set Representation for Taxonomy Expansion

ACL 2024findings

Taxonomy Expansion, which relies on modeling concepts and concept relations, can be formulated as a set representation learning task. The generalization of set, fuzzy set, incorporates uncertainty and measures the information within a semantic concept, making it suitable for concept modeling. Existi…

Cited by 4SourcePDFScholar
2024

LLM-Rec: Personalized Recommendation via Prompting Large Language Models

NAACL 2024findings

Text-based recommendation holds a wide range of practical applications due to its versatility, as textual descriptions can represent nearly any type of item. However, directly employing the original item descriptions may not yield optimal recommendation performance due to the lack of comprehensive i…

2024

RESPROMPT: Residual Connection Prompting Advances Multi-Step Reasoning in Large Language Models

NAACL 2024long

Chain-of-thought (CoT) has impressively unlocked the reasoning potential of large language models (LLMs). Yet, it falls short when tackling problems that require multiple reasoning steps. This limitation arises from the complex nature of multi-step reasoning processes: later stages often depend not…

2024

To the Globe (TTG): Towards Language-Driven Guaranteed Travel Planning

EMNLP 2024system demonstrations

Travel planning is a challenging and time-consuming task that aims to find an itinerary which satisfies multiple, interdependent constraints regarding flights, accommodations, attractions, and other travel arrangements. In this paper, we propose To the Globe (TTG), a real-time demo system that takes…

2023

CARE: Modeling Interacting Dynamics Under Temporal Environmental Variation

NeurIPS 2023poster

Modeling interacting dynamical systems, such as fluid dynamics and intermolecular interactions, is a fundamental research problem for understanding and simulating complex real-world systems. Many of these systems can be naturally represented by dynamic graphs, and graph neural network-based approach…

Cited by 11SourcePDFScholar
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…