Demo Data from SIGIR 2026 / reviewers in the wild / expert
Jingbo Shang
dblp:151/3145
· DBLP profile ↗
117ranked-venue papers
10as first author
82since 2021 · last 2026
0000-0002-7249-4404ORCID · corroborated
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 94 · 5 first-author · 71 since 2021Databases, data management, data science and information retrieval · 38 · 9 first-author · 14 since 2021Graphics, computer vision, multimedia, augmented reality and games · 8 · 7 since 2021Applied, interdisciplinary, general and emerging computing · 8 · 1 first-author · 3 since 2021Computer networks · 4 · 4 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
76 papers |
Language models and text generation · 26% Information extraction and text analysis · 22% Representation and self-supervised learning · 10% | |
| Databases, data mining, and information retrieval
27 papers |
Data mining · 62% Information retrieval · 12% Recommender systems · 9% | |
| Interdisciplinary, comprehensive, and emerging computing
4 papers |
Computing education · 60% Environmental and earth informatics · 17% Bioinformatics and computational biology · 9% |
Topics — the 30 heaviest of 198, each with the papers that count most for it
| Topic | Weight | Papers | Last | Evidence papers |
|---|---|---|---|---|
Natural language and speech › Information extraction and text analysis
text classification |
5.4 | 9 | 2024 | Text Grafting: Near-Distribution Weak Supervision for Minority Classes in Text Classification · EMNLP 2024 Incubating Text Classifiers Following User Instruction with Nothing but LLM · EMNLP 2024 Open-world Multi-label Text Classification with Extremely Weak Supervision · EMNLP 2024 |
Machine learning › Learning paradigms
weakly supervised learning |
4.9 | 8 | 2024 | Text Grafting: Near-Distribution Weak Supervision for Minority Classes in Text Classification · EMNLP 2024 Incubating Text Classifiers Following User Instruction with Nothing but LLM · EMNLP 2024 Open-world Multi-label Text Classification with Extremely Weak Supervision · EMNLP 2024 |
Natural language and speech › Language models and text generation
chain-of-thought reasoning |
2.6 | 4 | 2025 | OCEAN: Offline Chain-of-thought Evaluation and Alignment in Large Language Models · ICLR 2025 Chain-of-Table: Evolving Tables in the Reasoning Chain for Table Understanding · ICLR 2024 Can LLMs Learn from Previous Mistakes? Investigating LLMs' Errors to Boost for Reasoning · ACL (1) 2024 |
Natural language and speech › Language models and text generation
large language model |
2.5 | 4 | 2024 | Large Language Models for Time Series: A Survey · IJCAI 2024 Text Grafting: Near-Distribution Weak Supervision for Minority Classes in Text Classification · EMNLP 2024 Incubating Text Classifiers Following User Instruction with Nothing but LLM · EMNLP 2024 |
Machine learning › Trustworthy machine learning
interpretability |
2.1 | 3 | 2024 | Quantifying and Optimizing Global Faithfulness in Persona-driven Role-playing · NeurIPS 2024 Gradient-based Wang-Landau Algorithm: A Novel Sampler for Output Distribution of Neural Networks over the Input Space · ICML 2023 Learning Concise and Descriptive Attributes for Visual Recognition · ICCV 2023 |
Ubiquitous computing and smart environments › context recognition
activity recognition |
1.8 | 3 | 2025 | Matching Skeleton-based Activity Representations with Heterogeneous Signals for HAR · SenSys 2025 UniMTS: Unified Pre-training for Motion Time Series · NeurIPS 2024 UniTS: Short-Time Fourier Inspired Neural Networks for Sensory Time Series Classification · SenSys 2021 |
Natural language and speech › Language models and text generation › large language model › knowledge in language models
knowledge injection |
1.8 | 2 | 2026 | Bidirectional LMs are Better Knowledge Memorizers? A Benchmark for Real-world Knowledge Injection · ACL (1) 2026 MEMORYLLM: Towards Self-Updatable Large Language Models · ICML 2024 |
Machine learning › Representation and self-supervised learning
pre-training |
1.6 | 3 | 2024 | Fast-ELECTRA for Efficient Pre-training · ICLR 2024 Understand and Modularize Generator Optimization in ELECTRA-style Pretraining · ICML 2023 Text Is All You Need: Learning Language Representations for Sequential Recommendation · KDD 2023 |
Natural language and speech › Information extraction and text analysis
named entity recognition |
1.5 | 4 | 2021 | Weakly Supervised Named Entity Tagging with Learnable Logical Rules · ACL/IJCNLP (1) 2021 CrossWeigh: Training Named Entity Tagger from Imperfect Annotations · EMNLP/IJCNLP (1) 2019 Learning Named Entity Tagger using Domain-Specific Dictionary · EMNLP 2018 |
Data mining
clustering |
1.5 | 2 | 2025 | Contextual Inference From Sparse Shopping Transactions Based on Motif Patterns · IEEE Trans. Knowl. Data Eng. 2025 Goal-Driven Explainable Clustering via Language Descriptions · EMNLP 2023 |
Machine learning › Generative modeling
synthetic data generation |
1.5 | 2 | 2024 | Text Grafting: Near-Distribution Weak Supervision for Minority Classes in Text Classification · EMNLP 2024 Incubating Text Classifiers Following User Instruction with Nothing but LLM · EMNLP 2024 |
Machine learning › Representation and self-supervised learning
contrastive learning |
1.5 | 3 | 2024 | UniMTS: Unified Pre-training for Motion Time Series · NeurIPS 2024 Fine-grained Contrastive Learning for Relation Extraction · EMNLP 2022 UCTopic: Unsupervised Contrastive Learning for Phrase Representations and Topic Mining · ACL (1) 2022 |
Natural language and speech › Information extraction and text analysis › document understanding › legal text analysis
patent approval prediction |
1.3 | 2 | 2024 | Beyond Scaling: Predicting Patent Approval with Domain-specific Fine-grained Claim Dependency Graph · ACL (1) 2024 Towards Comprehensive Patent Approval Predictions: Beyond Traditional Document Classification · ACL (1) 2022 |
Knowledge, reasoning and agents › Knowledge representation and reasoning › knowledge acquisition › knowledge extraction
attribute discovery |
1.2 | 2 | 2023 | Towards Open-World Product Attribute Mining: A Lightly-Supervised Approach · ACL (1) 2023 OA-Mine: Open-World Attribute Mining for E-Commerce Products with Weak Supervision · WWW 2022 |
Natural language and speech › Information extraction and text analysis
relation extraction |
1.2 | 2 | 2023 | Open-world Semi-supervised Generalized Relation Discovery Aligned in a Real-world Setting · EMNLP 2023 Fine-grained Contrastive Learning for Relation Extraction · EMNLP 2022 |
Machine learning › Trustworthy machine learning
uncertainty estimation |
1.2 | 2 | 2023 | On Compositional Uncertainty Quantification for Seq2seq Graph Parsing · ICLR 2023 Neural-Symbolic Inference for Robust Autoregressive Graph Parsing via Compositional Uncertainty Quantification · EMNLP 2022 |
Data mining
pattern mining |
1.2 | 2 | 2025 | Contextual Inference From Sparse Shopping Transactions Based on Motif Patterns · IEEE Trans. Knowl. Data Eng. 2025 DPPred: An Effective Prediction Framework with Concise Discriminative Patterns · IEEE Trans. Knowl. Data Eng. 2018 |
Computer vision › Vision and language
multimodal reasoning |
1.0 | 1 | 2026 | SceneAlign: Aligning Multimodal Reasoning to Scene Graphs in Complex Visual Scenes · ACL (1) 2026 |
Natural language and speech › Language models and text generation › LLM agents
role-playing agents |
1.0 | 1 | 2026 | Deriving Character Logic from Storyline as Codified Decision Trees · ACL (1) 2026 |
Computer vision › Segmentation and scene understanding
scene graph |
1.0 | 1 | 2026 | SceneAlign: Aligning Multimodal Reasoning to Scene Graphs in Complex Visual Scenes · ACL (1) 2026 |
Data mining
text mining |
1.0 | 3 | 2019 | TextCube: Automated Construction and Multidimensional Exploration · Proc. VLDB Endow. 2019 Constructing and Mining Heterogeneous Information Networks from Massive Text · KDD 2019 Mining Quality Phrases from Massive Text Corpora · SIGMOD Conference 2015 |
Machine learning › Trustworthy machine learning
robustness |
1.0 | 2 | 2022 | Label Noise in Adversarial Training: A Novel Perspective to Study Robust Overfitting · NeurIPS 2022 CrossWeigh: Training Named Entity Tagger from Imperfect Annotations · EMNLP/IJCNLP (1) 2019 |
Machine learning › Representation and self-supervised learning › word representation
contextual representation |
0.9 | 2 | 2021 | "Average" Approximates "First Principal Component"? An Empirical Analysis on Representations from Neural Language Models · EMNLP (1) 2021 Contextualized Weak Supervision for Text Classification · ACL 2020 |
Data mining › text mining
information extraction and text analysis |
0.9 | 2 | 2021 | AutoName: A Corpus-Based Set Naming Framework · SIGIR 2021 Scientific Text Mining and Knowledge Graphs · KDD 2020 |
Natural language and speech › Language models and text generation
code generation |
0.9 | 1 | 2025 | LiveCodeBench Pro: How Do Olympiad Medalists Judge LLMs in Competitive Programming? · NeurIPS 2025 |
Natural language and speech › Language models and text generation
decoding |
0.9 | 1 | 2025 | Correlation and Navigation in the Vocabulary Key Representation Space of Language Models · ICLR 2025 |
Natural language and speech › Information extraction and text analysis › named entity processing
entity set expansion |
0.9 | 2 | 2020 | SynSetExpan: An Iterative Framework for Joint Entity Set Expansion and Synonym Discovery · EMNLP (1) 2020 Empower Entity Set Expansion via Language Model Probing · ACL 2020 |
Machine learning › Representation and self-supervised learning › representation matching
feature alignment |
0.9 | 1 | 2025 | Vector-ICL: In-context Learning with Continuous Vector Representations · ICLR 2025 |
Natural language and speech › Information extraction and text analysis › low-resource NLP › low-resource information extraction
few-shot information extraction |
0.9 | 1 | 2025 | Cuckoo: An IE Free Rider Hatched by Massive Nutrition in LLM's Nest · ACL (1) 2025 |
Natural language and speech › Language models and text generation › large language model reasoning
implicit reasoning |
0.9 | 1 | 2025 | Toward Multi-Session Personalized Conversation: A Large-Scale Dataset and Hierarchical Tree Framework for Implicit Reasoning · EMNLP 2025 |
Methods — techniques the papers use, named apart from their topics
contrastive learning · 4.2large language model · 3.1large language model prompting · 2.8fine-tuning · 2.8static analysis · 2.6graph neural network · 2.5in-context learning · 2.4clustering · 2.2transformer · 2.0verifier-guided regeneration · 1.7line-by-line failure analysis · 1.7large language model fine-tuning · 1.7benchmark annotation · 1.7language model pretraining · 1.6resource-aware scheduling · 1.5model aggregation · 1.5pre-training · 1.2distant supervision · 1.2
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2026 | Deriving Character Logic from Storyline as Codified Decision TreesabstractRole-playing (RP) agents rely on behavioral profiles to act consistently across diverse narrative contexts, yet existing profiles are largely unstructured, non-executable, and weakly validated, leading to brittle agent behavior.We propose Codified Decision Trees (CDT), a datadriven framework that induces an executable and interpretable decision structure from largescale narrative data.CDT represents behavioral profiles as a tree of conditional rules, where internal nodes correspond to validated scene conditions and leaves encode grounded behavioral statements, enabling deterministic retrieval of context-appropriate rules at execution time.The tree is learned by iteratively inducing candidate scene-action rules, validating them against data, and refining them through hierarchical specialization, yielding profiles that support transparent inspection and principled updates.Across multiple benchmarks, CDT substantially outperforms human-written profiles and prior profile induction methods on 85 characters across 16 artifacts, indicating that codified and validated behavioral representations lead to more reliable agent grounding.1 Letian Peng, Kun Zhou 0002, Longfei Yun, Yupeng Hou, Jingbo Shang |
ACL (1) | 5 |
| 2026 | SceneAlign: Aligning Multimodal Reasoning to Scene Graphs in Complex Visual ScenesabstractChuhan Wang, Xintong Li, Jennifer Yuntong Zhang, Junda Wu, Chengkai Huang, Lina Yao, Julian McAuley, Jingbo Shang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Xintong Li 0001, Jennifer Yuntong Zhang, Junda Wu, Chengkai Huang, Lina Yao 0001, Julian J. McAuley, Jingbo Shang |
ACL (1) | 8 |
| 2026 | Bidirectional LMs are Better Knowledge Memorizers? A Benchmark for Real-world Knowledge InjectionabstractYuwei Zhang, Wenhao Yu, Shangbin Feng, Yifan Zhu, Letian Peng, Jayanth Srinivasa, Gaowen Liu, Jingbo Shang. Proceedings of the 64th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2026. Yuwei Zhang 0001, Wenhao Yu 0002, Shangbin Feng, Letian Peng, Jayanth Srinivasa, Gaowen Liu, Jingbo Shang |
ACL (1) | 8 |
| 2025 | ZeroHAR: Sensor Context Augments Zero-Shot Wearable Action RecognitionabstractWearable Human Action Recognition (wHAR) uses motion sensor data to identify human movements, which is essential for mobile and wearable devices. However, traditional wHAR systems are only trained on a limited set of activities. Hence, they fail to generalize to diverse human motions, prompting Zero-Shot Learning (ZSL). Existing ZSL methods for wHAR focus solely on augmenting labels, such as representing them as attribute matrices, images, videos, or text. We propose ZeroHAR that enhances ZSL by not just focusing on activity labels, but by augmenting motion data with sensor context features. Our approach incorporates information about the sensor type, the Cartesian axis of the data, and the sensor's body position, providing the model with crucial spatial and biomechanical insights. This helps the model generalize better to new actions. First, we train the model by aligning the latent space of the motion time-series with its corresponding sensor context, while distancing it from unrelated sensor contexts. Finally, we train the model using the target activity descriptions. We tested our method against eight baselines on five benchmark HAR datasets with various sensors, placements, and activities. Our model shows exceptional generalizability across 18 motion time series classification benchmark datasets, outperforming the best baselines by 262% in the zero-shot setting. Ranak Roy Chowdhury, Ritvik Kapila, Ameya Panse, Xiyuan Zhang 0001, Diyan Teng, Rashmi Kulkarni, Dezhi Hong, Rajesh K. Gupta 0001, Jingbo Shang |
AAAI | 9 |
| 2025 | Cuckoo: An IE Free Rider Hatched by Massive Nutrition in LLM's NestabstractMassive high-quality data, both pre-training raw texts and post-training annotations, have been carefully prepared to incubate advanced large language models (LLMs).In contrast, for information extraction (IE), pre-training data, such as BIO-tagged sequences, are hard to scale up.We show that IE models can act as free riders on LLM resources by reframing next-token prediction into extraction for tokens already present in the context.Specifically, our proposed next tokens extraction (NTE) paradigm learns a versatile IE model, Cuckoo 1 , with 102.6M extractive data converted from LLM's pre-training and post-training data.Under the few-shot setting, Cuckoo adapts effectively to traditional and complex instruction-following IE with better performance than existing pretrained IE models.As a free rider, Cuckoo can naturally evolve with the ongoing advancements in LLM data preparation, benefiting from improvements in LLM training pipelines without additional manual effort.2 Letian Peng, Zilong Wang 0002, Jingbo Shang |
ACL (1) | 4 |
| 2025 | Direct Prompt Optimization with Continuous RepresentationsabstractPrompt optimization for language models faces challenges due to the large discrete search space, the reliance on continuous gradient updates, and the need to round continuous representations into discrete prompts, which causes inflexibility and instability.Existing methods attempt to address these by constraining the search space and adopting greedy, incremental improvements, but they often fail to fully leverage historical gradient information.In this paper, we model the prompt optimization problem by the probability distribution of the prompt and present a novel approach that integrates greedy strategies into optimization with continuous representations.This approach can exploit historical gradient information to address the instability caused by rounding in existing methods.Our study indicates that using continuous representations can improve prompt optimization performance on both text classification and attack tasks, as well as models, including GPT-2, OPT, Vicuna, and LLaMA-2, and also be adaptable to models of different sizes. Yangkun Wang, Zihan Wang 0001, Jingbo Shang |
ACL (1) | 3 |
| 2025 | Self-Taught Agentic Long Context UnderstandingabstractYufan Zhuang, Xiaodong Yu, Jialian Wu, Ximeng Sun, Ze Wang, Jiang Liu, Yusheng Su, Jingbo Shang, Zicheng Liu, Emad Barsoum. Proceedings of the 63rd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2025. Yufan Zhuang, Jialian Wu, Ximeng Sun, Ze Wang 0008, Jiang Liu 0014, Yusheng Su, Jingbo Shang, Zicheng Liu 0001, Emad Barsoum |
ACL (1) | 8 |
| 2025 | VeriLocc: End-to-End Cross-Architecture Register Allocation via LLMabstractModern GPUs evolve rapidly, yet production compilers still rely on hand-crafted register allocation heuristics that require substantial retuning for each hardware generation.We introduce VERILOCC, a framework that combines large language models (LLMs) with formal compiler techniques to enable generalizable and verifiable register allocation across GPU architectures.VERILOCC fine-tunes an LLM to translate intermediate representations (MIRs) into target-specific register assignments, aided by static analysis for cross-architecture normalization and generalization and a verifierguided regeneration loop to ensure correctness.Evaluated on matrix multiplication (GEMM) and multi-head attention (MHA), VERILOCC achieves 85-99% single-shot accuracy and near-100% [email protected] study shows that VERILOCC discovers more performant assignments than expert-tuned libraries, outperforming rocBLAS by over 10% in runtime. Lesheng Jin, Zhenyuan Ruan, Haohui Mai, Jingbo Shang |
EMNLP | 4 |
| 2025 | Toward Multi-Session Personalized Conversation: A Large-Scale Dataset and Hierarchical Tree Framework for Implicit ReasoningabstractThere has been a surge in the use of large language models (LLM) conversational agents to generate responses based on long-term history from multiple sessions.However, existing long-term open-domain dialogue datasets lack complex, real-world personalization and fail to capture implicit reasoning-where relevant information is embedded in subtle, syntactic, or semantically distant connections rather than explicit statements.In such cases, traditional retrieval methods fail to capture relevant context, and long-context modeling also becomes inefficient due to numerous complicated personarelated details.To address this gap, we introduce IMPLEXCONV, a large-scale long-term dataset with 2,500 examples, each containing approximately 100 conversation sessions, designed to study implicit reasoning in personalized dialogues.Additionally, we propose TAC-ITREE, a novel hierarchical tree framework that structures conversation history into multiple levels of summarization.Instead of brute-force searching all data, TACITREE enables an efficient, level-based retrieval process where models refine their search by progressively selecting relevant details.Our experiments demonstrate that TACITREE significantly improves the ability of LLMs to reason over long-term conversations with implicit contextual dependencies.I love playing sports like basketball and swimming.I'm going on a sports stadium tour!Aug 15, 2024 I broke my leg in a car accident... Xintong Li 0001, Jalend Bantupalli, Ria Dharmani, Yuwei Zhang 0001, Jingbo Shang |
EMNLP | 5 |
| 2025 | CoMMIT: Coordinated Multimodal Instruction TuningabstractXintong Li, Junda Wu, Tong Yu, Rui Wang, Yu Wang, Xiang Chen, Jiuxiang Gu, Lina Yao, Julian McAuley, Jingbo Shang. Proceedings of the 2025 Conference on Empirical Methods in Natural Language Processing. 2025. Xintong Li 0001, Junda Wu, Tong Yu 0001, Rui Wang 0088, Xiang Chen 0010, Jiuxiang Gu, Lina Yao 0001, Julian J. McAuley, Jingbo Shang |
EMNLP | 10 |
| 2025 | Speculative RAG: Enhancing Retrieval Augmented Generation through DraftingabstractRetrieval augmented generation (RAG) combines the generative abilities of large language models (LLMs) with external knowledge sources to provide more accurate and up-to-date responses. Recent RAG advancements focus on improving retrieval outcomes through iterative LLM refinement or self-critique capabilities acquired through additional instruction tuning of LLMs. In this work, we introduce Speculative RAG - a framework that leverages a larger generalist LM to efficiently verify multiple RAG drafts produced in parallel by a smaller, distilled specialist LM. Each draft is generated from a distinct subset of retrieved documents, offering diverse perspectives on the evidence while reducing input token counts per draft. This approach enhances comprehension of each subset and mitigates potential position bias over long context. Our method accelerates RAG by delegating drafting to the smaller specialist LM, with the larger generalist LM performing a single verification pass over the drafts. Extensive experiments demonstrate that Speculative RAG achieves state-of-the-art performance with reduced latency on TriviaQA, MuSiQue, PopQA, PubHealth, and ARC-Challenge benchmarks. It notably enhances accuracy by up to 12.97% while reducing latency by 50.83% compared to conventional RAG systems on PubHealth. Zilong Wang 0002, Zifeng Wang 0002, Long T. Le, Huaixiu Steven Zheng, Swaroop Mishra, Vincent Perot, Yuwei Zhang 0001, Anush Mattapalli, Ankur Taly, Jingbo Shang, Chen-Yu Lee, Tomas Pfister |
ICLR | 10 |
| 2025 | Correlation and Navigation in the Vocabulary Key Representation Space of Language ModelsabstractLanguage model (LM) decoding is based on the next-token prediction (NTP) probability distribution. For neural LMs (e.g., Transformer-based), NTP distribution is
essentially a softmax-regularized dot product between an encoded input context
(query) and fixed vocabulary representations (keys). In this paper, we study the
effect of the key distribution on the NTP distribution, with a focus on whether
the similarity between keys will trigger spurious correlations in NTP. Through
knowledge-probing tasks, we show that in the NTP distribution, the few top-ranked
tokens are typically accurate. However, the middle-ranked prediction is highly biased
towards the tokens that are distributionally (not necessarily semantically) similar to
these top ones. For instance, if “P” is predicted as the top-1 token, “A”-“Z” will all
be ranked high in NTP, no matter whether they can lead to correct decoding results.
This hurts the sampling diversity and makes the sampling of correct, long-tail
results hopeless and noisy. We attempt to alleviate this issue via a novel in-context
method that iteratively pushes the query representation away from explored regions.
Specifically, we include the explored decoding results in the context and prompt
the LM to generate something else, which encourages the LM to produce a query
representation that has small dot products with explored keys. Experiments on
knowledge-probing tasks show that our method leads to efficient navigation away
from explored keys to correct new keys. We further extend our method to open-ended and chain-of-thought (for reasoning) generation. Experiment results show
that ICN contributes to better generation diversity and improved self-consistency
voting performance. Finally, we discuss potential training issues caused by the
fixed key space together with the challenges and possible ways to address them in
future research. Letian Peng, Chenyang An, Jingbo Shang |
ICLR | 3 |
| 2025 | OCEAN: Offline Chain-of-thought Evaluation and Alignment in Large Language ModelsabstractOffline evaluation of LLMs is crucial in understanding their capacities, though current methods remain underexplored in existing research. In this work, we focus on the offline evaluation of the chain-of-thought capabilities and show how to optimize LLMs based on the proposed evaluation method. To enable offline feedback with rich knowledge and reasoning paths, we use knowledge graphs (KGs) (e.g., Wikidata5M) to provide feedback on the generated chain of thoughts. Due to the heterogeneity between LLM reasoning and KG structures, direct interaction and feedback from knowledge graphs on LLM behavior are challenging, as they require accurate entity linking and grounding of LLM-generated chains of thought in the KG. To address the above challenge, we propose an offline chain-of-thought evaluation framework, OCEAN, which models chain-of-thought reasoning in LLMs as a Markov Decision Process (MDP), and evaluate the policy’s alignment with KG preference modeling. To overcome the reasoning heterogeneity and grounding problems, we leverage on-policy KG exploration and reinforcement learning to model a KG policy that generates token-level likelihood distributions for LLM-generated chain-of-thought reasoning paths, simulating KG reasoning preference. Then we incorporate the knowledge-graph feedback on the validity and alignment of the generated reasoning paths into inverse propensity scores and propose KG-IPS estimator. Theoretically, we prove the unbiasedness of the proposed KG-IPS estimator and provide a lower bound on its variance. With the off-policy evaluated value function, we can directly enable off-policy optimization to further enhance chain-of-thought alignment. Our empirical study shows that OCEAN can be efficiently optimized for generating chain-of-thought reasoning paths with higher estimated values without affecting LLMs’ general abilities in downstream tasks or their internal knowledge. Junda Wu, Xintong Li 0001, Ruoyu Wang 0038, Yu Xia 0007, Yuxin Xiong, Jianing Wang 0002, Tong Yu 0001, Xiang Chen 0010, Branislav Kveton, Lina Yao 0001, Jingbo Shang, Julian J. McAuley |
ICLR | 11 |
| 2025 | Vector-ICL: In-context Learning with Continuous Vector RepresentationsabstractLarge language models (LLMs) have shown remarkable in-context learning (ICL) capabilities on textual data. We explore whether these capabilities can be extended to continuous vectors from diverse domains, obtained from black-box pretrained encoders. By aligning input data with an LLM's embedding space through lightweight projectors, we observe that LLMs can effectively process and learn from these projected vectors, which we term Vector-ICL. In particular, we find that pretraining projectors with general language modeling objectives enables Vector-ICL, while task-specific finetuning further enhances performance. In our experiments across various tasks and modalities, including text reconstruction, numerical function regression, text classification, summarization, molecule captioning, time-series classification, graph classification, and fMRI decoding, Vector-ICL often surpasses both few-shot ICL and domain-specific model or tuning. We further conduct analyses and case studies, indicating the potential of LLMs to process vector representations beyond traditional token-based paradigms. Yufan Zhuang, Chandan Singh, Jingbo Shang, Jianfeng Gao 0001 |
ICLR | 4 |
| 2025 | Entangled Relations: Leveraging NLI and Meta-analysis to Enhance Biomedical Relation ExtractionabstractWilliam P Hogan, Jingbo Shang. Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2025. William Hogan, Jingbo Shang |
NAACL (Long Papers) | 2 |
| 2025 | ALERT: An LLM-powered Benchmark for Automatic Evaluation of Recommendation ExplanationsabstractYichuan Li, Xinyang Zhang, Chenwei Zhang, Mao Li, Tianyi Liu, Pei Chen, Yifan Gao, Kyumin Lee, Kaize Ding, Zhengyang Wang, Zhihan Zhang, Jingbo Shang, Xian Li, Trishul Chilimbi. Proceedings of the 2025 Conference of the Nations of the Americas Chapter of the Association for Computational Linguistics: Human Language Technologies (Volume 1: Long Papers). 2025. Yichuan Li 0001, Xinyang Zhang 0002, Yifan Gao 0001, Kyumin Lee, Kaize Ding, Zhihan Zhang 0001, Jingbo Shang, Trishul Chilimbi |
NAACL (Long Papers) | 12 |
| 2025 | Codifying Character Logic in Role-PlayingabstractThis paper introduces Codified Profiles for role-playing, a novel approach that represents character logic as structured, executable functions for behavioral decision-making.
Converted by large language model (LLM) from textual profiles, each codified profile defines a set of functions parse_by_scene(scene) that output multiple logic-grounded assertions according to scene, using both explicit control structures (e.g., if-then-else) and flexible check_condition(scene, question) functions where each question is a semantically meaningful prompt about the scene (e.g., "Is the character in danger?") discriminated by the role-playing LLM as true, false, or unknown.
This explicit representation offers three key advantages over traditional prompt-based textual profiles, which append character descriptions directly into text prompts:
(1) Persistence, by enforcing complete and consistent execution of character logic, rather than relying on the model's implicit reasoning;
(2) Updatability, through systematic inspection and revision of behavioral logic, which is difficult to track or debug in prompt-only approaches;
(3) Controllable Randomness, by supporting stochastic behavior directly within the logic, enabling fine-grained variability that prompting alone struggles to achieve.
To validate these advantages, we introduce a new benchmark constructed from 83 characters and 5,141 scenes curated from Fandom, using natural language inference (NLI)-based scoring to compare character responses against ground-truths.
Our experiments demonstrate the significant benefits of codified profiles in improving persistence, updatability, and behavioral diversity.
Notably, by offloading a significant portion of reasoning to preprocessing, codified profiles enable even 1B-parameter models to perform high-quality role-playing, providing an efficient, lightweight foundation for local deployment of role-play agents. Letian Peng, Jingbo Shang |
NeurIPS | 2 |
| 2025 | Training Language Models to Generate Quality Code with Program Analysis FeedbackabstractCode generation with large language models (LLMs), often termed vibe coding, is increasingly adopted in production but fails to ensure code quality, particularly in security (e.g., SQL injection vulnerabilities) and maintainability (e.g., missing type annotations). Existing methods, such as supervised fine-tuning and rule-based post-processing, rely on labor-intensive annotations or brittle heuristics, limiting their scalability and effectiveness. We propose REAL (Reinforcement rEwards from Automated anaLysis), a reinforcement learning framework that trains LLMs to generate production-quality code using program analysis–guided feedback. Specifically, REAL integrates two automated signals: (1) static analyzers detecting security and maintainability defects and (2) unit tests ensuring functional correctness. Unlike prior work, our framework is prompt-agnostic and reference-free, enabling scalable supervision without manual intervention. Experiments across multiple datasets and model scales demonstrate that REAL outperforms state-of-the-art methods in simultaneous assessments of functionality and code quality. Our work bridges the gap between rapid prototyping and production-ready code, enabling LLMs to deliver both speed and quality. Zilong Wang 0002, Junxia Cui, Xiaohan Fu, Haohui Mai, Viswanathan Krishnan, Jianfeng Gao 0001, Jingbo Shang |
NeurIPS | 10 |
| 2025 | LiveCodeBench Pro: How Do Olympiad Medalists Judge LLMs in Competitive Programming?abstractRecent reports claim that large language models (LLMs) now outperform elite humans in competitive programming. Drawing on knowledge from a group of medalists in international algorithmic contests, we revisit this claim, examining how LLMs differ from human experts and where limitations still remain. We introduce LiveCodeBench Pro, a benchmark composed of problems from Codeforces, ICPC, and IOI that are continuously updated to reduce the likelihood of data contamination. A team of Olympiad medalists annotates every problem for algorithmic categories and conducts a line-by-line analysis of failed model-generated submissions. Using this new data and benchmark, we find that frontier models still have significant limitations: without external tools, the best model achieves only 53\% pass@1 on medium-difficulty problems and 0\% on hard problems, domains where expert humans still excel. We also find that LLMs succeed at implementation-heavy problems but struggle with nuanced algorithmic reasoning and complex case analysis, often generating confidently incorrect justifications. High performance appears largely driven by implementation precision and tool augmentation, not superior reasoning. LiveCodeBench Pro thus highlights the significant gap to human grandmaster levels, while offering fine-grained diagnostics to steer future improvements in code-centric LLM reasoning. Zihan Zheng, Zerui Cheng, Shang Zhou, Hansen He, Dongruixuan Li, Stanley Wei, Hangyi Hao, Jianzhu Yao, Peiyao Sheng, Zixuan Wang 0029, Wenhao Chai, Aleksandra Korolova, Peter Henderson 0002, Sanjeev Arora, Pramod Viswanath, Jingbo Shang, Saining Xie |
NeurIPS | 18 |
| 2025 | Mixture of Inputs: Text Generation Beyond Discrete Token SamplingabstractIn standard autoregressive generation, an LLM predicts the next-token distribution, samples a discrete token, and then discards the distribution, passing only the sampled token as new input. To preserve this distribution’s rich information, we propose Mixture of Inputs (MoI), a training-free method for autoregressive generation. After generating a token following the standard paradigm, we construct a new input that blends the generated discrete token with the previously discarded token distribution. Specifically, we employ a Bayesian estimation method that treats the token distribution as the prior, the sampled token as the observation, and replaces the conventional one-hot vector with the continuous posterior expectation as the new model input. MoI allows the model to maintain a richer internal representation throughout the generation process, resulting in improved text quality and reasoning capabilities. On mathematical reasoning, code generation, and PhD-level QA tasks, MoI consistently improves performance across multiple models including QwQ-32B, Nemotron-Super-49B, Gemma-3-27B, and DAPO-Qwen-32B, with no additional training and negligible computational overhead. Yufan Zhuang, Chandan Singh, Jingbo Shang, Jianfeng Gao 0001 |
NeurIPS | 4 |
| 2025 | Matching Skeleton-based Activity Representations with Heterogeneous Signals for HARabstractIn human activity recognition (HAR), activity labels have typically been encoded in one-hot format, which has a recent shift towards using textual representations to provide contextual knowledge. Here, we argue that HAR should be anchored to physical motion data, as motion forms the basis of activity and applies effectively across sensing systems, whereas text is inherently limited. We propose SKELAR, a novel HAR framework that pretrains activity representations from skeleton data and matches them with heterogeneous HAR signals. Our method addresses two major challenges: (1) capturing core motion knowledge without context-specific details. We achieve this through a self-supervised coarse angle reconstruction task that recovers joint rotation angles, invariant to both users and deployments; (2) adapting the representations to downstream tasks with varying modalities and focuses. To address this, we introduce a self-attention matching module that dynamically prioritizes relevant body parts in a data-driven manner. Given the lack of corresponding labels in existing skeleton data, we establish MASD, a new HAR dataset with IMU, WiFi, and skeleton, collected from 20 subjects performing 27 activities. This is the first broadly applicable HAR dataset with time-synchronized data across three modalities. Experiments show that SKELAR achieves the state-of-the-art performance in both full-shot and few-shot settings. We also demonstrate that SKELAR can effectively leverage synthetic skeleton data to extend its use in scenarios without skeleton collections. Shuheng Li, Jiayun Zhang, Xiaohan Fu, Xiyuan Zhang 0001, Jingbo Shang, Rajesh K. Gupta 0001 |
SenSys | 5 |
| 2025 | Contextual Inference From Sparse Shopping Transactions Based on Motif PatternsabstractInferring contextual information such as demographics from historical transactions is valuable to public agencies and businesses. Existing methods are data-hungry and do not work well when the available records of transactions are sparse. We consider here specifically inference of demographic information using limited historical grocery transactions from a few random trips that a typical business or public service organization may see. We propose a novel method calledDemoMotifto build a network model from heterogeneous data and identify subgraph patterns (i.e., motifs) that enable us to infer demographic attributes. We then design a novel motif context selection algorithm to find specific node combinations significant to certain demographic groups. Finally, we learn representations of households using these selected motif instances as context, and employ a standard classifier (e.g., SVM) for inference. For evaluation purposes, we use three real-world consumer datasets, spanning different regions and time periods in the U.S. We evaluate the framework for predicting three attributes: ethnicity, seniority of household heads, and presence of children. Extensive experiments and case studies demonstrate thatDemoMotifis capable of inferring household demographics using only a small number (e.g., fewer than 10) of random grocery trips, significantly outperforming the state-of-the-art. Jiayun Zhang, Xinyang Zhang 0002, Dezhi Hong, Rajesh K. Gupta 0001, Jingbo Shang |
IEEE Trans. Knowl. Data Eng. | 5 |
| 2024 | Learn from Failure: Fine-tuning LLMs with Trial-and-Error Data for Intuitionistic Propositional Logic ProvingabstractChenyang An, Zhibo Chen, Qihao Ye, Emily First, Letian Peng, Jiayun Zhang, Zihan Wang, Sorin Lerner, Jingbo Shang. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Chenyang An, Zhibo Chen 0009, Qihao Ye, Emily First, Letian Peng, Jiayun Zhang, Zihan Wang 0001, Sorin Lerner, Jingbo Shang |
ACL (1) | 9 |
| 2024 | Beyond Scaling: Predicting Patent Approval with Domain-specific Fine-grained Claim Dependency GraphabstractXiaochen Gao, Feng Yao, Kewen Zhao, Beilei He, Animesh Kumar, Vish Krishnan, Jingbo Shang. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Xiaochen Gao, Kewen Zhao, Beilei He, Animesh Kumar, Vish Krishnan, Jingbo Shang |
ACL (1) | 7 |
| 2024 | Answer is All You Need: Instruction-following Text Embedding via Answering the QuestionabstractLetian Peng, Yuwei Zhang, Zilong Wang, Jayanth Srinivasa, Gaowen Liu, Zihan Wang, Jingbo Shang. Proceedings of the 62nd Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2024. Letian Peng, Yuwei Zhang 0001, Zilong Wang 0002, Jayanth Srinivasa, Gaowen Liu, Zihan Wang 0001, Jingbo Shang |
ACL (1) | 7 |
| 2024 | Can LLMs Learn from Previous Mistakes? Investigating LLMs' Errors to Boost for ReasoningabstractLarge language models (LLMs) have demonstrated striking reasoning capability.Recent works have shown the benefits to LLMs from fine-tuning golden-standard Chain-of-Thought (CoT) rationales or using them as correct examples in few-shot prompting.While humans can indeed imitate correct examples, learning from our mistakes is another vital aspect of human cognition.Hence, a question naturally arises: can LLMs learn and benefit from their mistakes, especially for their reasoning?This study investigates this problem from both the prompting and model-tuning perspectives.We begin by introducing COTERRORSET, a new benchmark with 558,960 questions, each designed with both correct and error references, and demonstrating the types and reasons for making such mistakes.To explore the effectiveness of those mistakes, we design two methods:(1) Self-rethinking prompting guides LLMs to rethink whether they have made similar previous mistakes; and (2) Mistake tuning involves finetuning models in both correct and incorrect reasoning domains, rather than only tuning models to learn ground truth in traditional methodology.We conduct a series of experiments to prove LLMs can obtain benefits from mistakes in both directions.Our two methods offer potentially cost-effective strategies by leveraging errors to enhance reasoning capabilities, which costs significantly less than creating meticulously hand-crafted golden references.We ultimately make a thorough analysis of the reasons behind LLMs' errors, which provides directions that future research needs to overcome.COTERRORSET will be published soon on https://github.com/YookiTong/ Learn-from-Mistakes-CotErrorSet. Yongqi Tong, Jingbo Shang |
ACL (1) | 6 |
| 2024 | Towards Few-shot Entity Recognition in Document Images: A Graph Neural Network Approach Robust to Image ManipulationabstractRecent advances of incorporating layout information, typically bounding box coordinates, into pre-trained language models have achieved significant performance in entity recognition from document images. Using coordinates can easily model the position of each token, but they are sensitive to manipulations in document images (e.g., shifting, rotation or scaling) which are common in real scenarios. Such limitation becomes even worse when the training data is limited in few-shot settings. In this paper, we propose a novel framework, LAGER, which leverages the topological adjacency relationship among the tokens through learning their relative layout information with graph neural networks. Specifically, we consider the tokens in the documents as nodes and formulate the edges based on the topological heuristics. Such adjacency graphs are invariant to affine transformations, making it robust to the common image manipulations. We incorporate these graphs into the pre-trained language model by adding graph neural network layers on top of the language model embeddings. Extensive experiments on two benchmark datasets show that LAGER significantly outperforms strong baselines under different few-shot settings and also demonstrate better robustness to manipulations. Prashant Krishnan, Zilong Wang 0002, Yangkun Wang, Jingbo Shang |
LREC/COLING | 4 |
| 2024 | Open-world Multi-label Text Classification with Extremely Weak SupervisionabstractWe study open-world multi-label text classification under extremely weak supervision (XWS), where the user only provides a brief description for classification objectives without any labels or ground-truth label space.Similar single-label XWS settings have been explored recently, however, these methods cannot be easily adapted for multi-label.We observe that (1) most documents have a dominant class covering the majority of content and (2) long-tail labels would appear in some documents as a dominant class.Therefore, we first utilize the user description to prompt a large language model (LLM) for dominant keyphrases of a subset of raw documents, and then construct a (initial) label space via clustering.We further apply a zero-shot multi-label classifier to locate the documents with small top predicted scores, so we can revisit their dominant keyphrases for more long-tail labels.We iterate this process to discover a comprehensive label space and construct a multi-label classifier as a novel method, X-MLClass.X-MLClass exhibits a remarkable increase in ground-truth label space coverage on various datasets, for example, a 40% improvement on the AAPD dataset over topic modeling and keyword extraction methods.Moreover, X-MLClass achieves the best end-to-end multi-label classification accuracy. B Prompt Templates for Generating KeyphrasesCode 2 provides an example of the prompt used to generate keyphrases for a selected chunk of the Amazon-531 dataset.Users can help us define the objective with examples.For example, the coarse-grained objectives look like "games" and "animals", while the corresponding fine-grained objectives are "trading_card_games" and "reptiles". Xintong Li 0001, Jinya Jiang, Ria Dharmani, Jayanth Srinivasa, Gaowen Liu, Jingbo Shang |
EMNLP | 6 |
| 2024 | Incubating Text Classifiers Following User Instruction with Nothing but LLMabstractIn this paper, we aim to generate text classification data given arbitrary class definitions (i.e., user instruction), so one can train a text classifier without any human annotation or raw corpus.Recent advances in large language models (LLMs) lead to pioneer attempts to individually generate texts for each class via prompting.In this paper, we propose Incubator, the first framework that can handle complicated and even mutually dependent classes (e.g., "TED Talk given by Educator" and "Other").Specifically, our Incubator is a fine-tuned LLM that takes the instruction of all class definitions as input, and in each inference, it can jointly generate one sample for every class.First, we tune Incubator on the instruction-to-data mappings that we obtained from classification datasets and descriptions on Hugging Face together with in-context augmentation by GPT-4.To emphasize the uniformity and diversity in generations, we refine Incubator by fine-tuning with the cluster centers of semantic textual embeddings of the generated samples.We compare Incubator on various classification tasks with strong baselines such as direct LLM-based inference and training data generation by prompt engineering.Experiments show Incubator is able to (1) outperform previous methods on traditional benchmarks, (2) take label interdependency and user preference into consideration, and (3) enable logical text mining by incubating multiple classifiers. Letian Peng, Zilong Wang 0002, Jingbo Shang |
EMNLP | 3 |
| 2024 | Text Grafting: Near-Distribution Weak Supervision for Minority Classes in Text ClassificationabstractFor extremely weak-supervised text classification, pioneer research generates pseudo labels by mining texts similar to the class names from the raw corpus, which may end up with very limited or even no samples for the minority classes.Recent works have started to generate the relevant texts by prompting LLMs using the class names or definitions; however, there is a high risk that LLMs cannot generate indistribution (i.e., similar to the corpus where the text classifier will be applied) data, leading to ungeneralizable classifiers.In this paper, we combine the advantages of these two approaches and propose to bridge the gap via a novel framework, text grafting, which aims to obtain clean and near-distribution weak supervision for minority classes.Specifically, we first use LLM-based logits to mine masked templates from the raw corpus, which have a high potential for data synthesis into the target minority class.Then, the templates are filled by state-of-the-art LLMs to synthesize neardistribution texts falling into minority classes.Text grafting shows significant improvement over direct mining or synthesis on minority classes.We also use analysis and case studies to comprehend the property of text grafting. Letian Peng, Yi Gu 0002, Chengyu Dong, Zihan Wang 0001, Jingbo Shang |
EMNLP | 5 |
| 2024 | Data Contamination Can Cross Language BarriersabstractThe opacity in developing large language models (LLMs) is raising growing concerns about the potential contamination of public benchmarks in the pre-training data.Existing contamination detection methods are typically based on the text overlap between training and evaluation data, which can be too superficial to reflect deeper forms of contamination.In this paper, we first present a cross-lingual form of contamination that inflates LLMs' performance while evading current detection methods, deliberately injected by overfitting LLMs on the translated versions of benchmark test sets.Then, we propose generalization-based approaches to unmask such deeply concealed contamination.Specifically, we examine the LLM's performance change after modifying the original benchmark by replacing the false answer choices with correct ones from other questions.Contaminated models can hardly generalize to such easier situations, where the false choices can be not even wrong, as all choices are correct in their memorization.Experimental results demonstrate that cross-lingual contamination can easily fool existing detection methods, but not ours.In addition, we discuss the potential utilization of cross-lingual contamination in interpreting LLMs' working mechanisms and in post-training LLMs for enhanced multilingual capabilities.The code and dataset we use can be obtained from https:// github.com/ShangDataLab/Deep-Contam. * Equal contribution. Listing order is random.Para Sócrates, el alma se daña por la falta de __A__.A. conocimiento, B. riqueza C. comunidad, D. coraje For Socrates, the soul is harmed by lack of _A_. A. knowledge, B. wealth C. Yufan Zhuang, Sunan Xu, Animesh Kumar, Jingbo Shang |
EMNLP | 6 |
| 2024 | Chain-of-Table: Evolving Tables in the Reasoning Chain for Table UnderstandingabstractTable-based reasoning with large language models (LLMs) is a promising direction to tackle many table understanding tasks, such as table-based question answering and fact verification. Compared with generic reasoning, table-based reasoning requires the extraction of underlying semantics from both free-form questions and semi-structured tabular data. Chain-of-Thought and its similar approaches incorporate the reasoning chain in the form of textual context, but it is still an open question how to effectively leverage tabular data in the reasoning chain. We propose the Chain-of-Table framework, where tabular data is explicitly used in the reasoning chain as a proxy for intermediate thoughts. Specifically, we guide LLMs using in-context learning to iteratively generate operations and update the table to represent a tabular reasoning chain. LLMs can therefore dynamically plan the next operation based on the results of the previous ones. This continuous evolution of the table forms a chain, showing the reasoning process for a given tabular problem. The chain carries structured information of the intermediate results, enabling more accurate and reliable predictions. Chain-of-Table achieves new state-of-the-art performance on WikiTQ, FeTaQA, and TabFact benchmarks across multiple LLM choices. Zilong Wang 0002, Chun-Liang Li, Julian Martin Eisenschlos, Vincent Perot, Zifeng Wang 0002, Lesly Miculicich, Yasuhisa Fujii, Jingbo Shang, Chen-Yu Lee, Tomas Pfister |
ICLR | 9 |
| 2024 | Fast-ELECTRA for Efficient Pre-trainingabstractELECTRA pre-trains language models by detecting tokens in a sequence that have been replaced by an auxiliary model. Although ELECTRA offers a significant boost in efficiency, its potential is constrained by the training cost brought by the auxiliary model. Notably, this model, which is jointly trained with the main model, only serves to assist the training of the main model and is discarded post-training. This results in a substantial amount of training cost being expended in vain. To mitigate this issue, we propose Fast-ELECTRA, which leverages an existing language model as the auxiliary model. To construct a learning curriculum for the main model, we smooth its output distribution via temperature scaling following a descending schedule. Our approach rivals the performance of state-of-the-art ELECTRA-style pre-training methods, while significantly eliminating the computation and memory cost brought by the joint training of the auxiliary model. Our method also reduces the sensitivity to hyper-parameters and enhances the pre-training stability. Chengyu Dong, Hao Cheng 0002, Jingbo Shang, Jianfeng Gao 0001, Xiaodong Liu 0003 |
ICLR | 4 |
| 2024 | Toward Student-oriented Teacher Network Training for Knowledge DistillationabstractHow to conduct teacher training for knowledge distillation is still an open problem. It has been widely observed that a best-performing teacher does not necessarily yield the best-performing student, suggesting a fundamental discrepancy between the current teacher training practice and the ideal teacher training strategy. To fill this gap, we explore the feasibility of training a teacher that is oriented toward student performance with empirical risk minimization (ERM). Our analyses are inspired by the recent findings that the effectiveness of knowledge distillation hinges on the teacher’s capability to approximate the true label distribution of training inputs. We theoretically establish that ERM minimizer can approximate the true label distribution of training data as long as the feature extractor of the learner network is Lipschitz continuous and is robust to feature transformations. In light of our theory, we propose a teacher training method SoTeacher which incorporates Lipschitz regularization and consistency regularization into ERM. Experiments on benchmark datasets using various knowledge distillation algorithms and teacher-student pairs confirm that SoTeacher can improve student accuracy consistently. Chengyu Dong, Jingbo Shang |
ICLR | 3 |
| 2024 | MEMORYLLM: Towards Self-Updatable Large Language ModelsabstractExisting Large Language Models (LLMs) usually remain static after deployment, which might make it hard to inject new knowledge into the model. We aim to build models containing a considerable portion of self-updatable parameters, enabling the model to integrate new knowledge effectively and efficiently. To this end, we introduce MEMORYLLM, a model that comprises a transformer and a fixed-size memory pool within the latent space of the transformer. MEMORYLLM can self-update with text knowledge and memorize the knowledge injected earlier. Our evaluations demonstrate the ability of MEMORYLLM to effectively incorporate new knowledge, as evidenced by its performance on model editing benchmarks. Meanwhile, the model exhibits long-term information retention capacity, which is validated through our custom-designed evaluations and long-context benchmarks. MEMORYLLM also shows operational integrity without any sign of performance degradation even after nearly a million memory updates. Our code and model are open-sourced at https://github.com/wangyu-ustc/MemoryLLM. Yu Wang 0170, Yifan Gao 0001, Xiusi Chen, Haoming Jiang, Jingfeng Yang 0001, Qingyu Yin, Zheng Li 0018, Jingbo Shang, Julian J. McAuley |
ICML | 11 |
| 2024 | Large Language Models for Time Series: A Survey
Xiyuan Zhang 0001, Ranak Roy Chowdhury, Rajesh K. Gupta 0001, Jingbo Shang |
IJCAI | 4 |
| 2024 | UniMTS: Unified Pre-training for Motion Time SeriesabstractMotion time series collected from low-power, always-on mobile and wearable devices such as smartphones and smartwatches offer significant insights into human behavioral patterns, with wide applications in healthcare, automation, IoT, and AR/XR. However, given security and privacy concerns, building large-scale motion time series datasets remains difficult, hindering the development of pre-trained models for human activity analysis. Typically, existing models are trained and tested on the same dataset, leading to poor generalizability across variations in device location, device mounting orientation, and human activity type. In this paper, we introduce UniMTS, the first unified pre-training procedure for motion time series that generalizes across diverse device latent factors and activities. Specifically, we employ a contrastive learning framework that aligns motion time series with text descriptions enriched by large language models. This helps the model learn the semantics of time series to generalize across activities. Given the absence of large-scale motion time series data, we derive and synthesize time series from existing motion skeleton data with all-joint coverage. We use spatio-temporal graph networks to capture the relationships across joints for generalization across different device locations. We further design rotation-invariant augmentation to make the model agnostic to changes in device mounting orientations. Our model shows exceptional generalizability across 18 motion time series classification benchmark datasets, outperforming the best baselines by 340% in the zero-shot setting, 16.3% in the few-shot setting, and 9.2% in the full-shot setting. Xiyuan Zhang 0001, Diyan Teng, Ranak Roy Chowdhury, Shuheng Li, Dezhi Hong, Rajesh K. Gupta 0001, Jingbo Shang |
NeurIPS | 7 |
| 2024 | Quantifying and Optimizing Global Faithfulness in Persona-driven Role-playingabstractPersona-driven role-playing (PRP) aims to build AI characters that can respond to user queries by faithfully sticking with \emph{all} (factual) statements in persona documents.
Unfortunately, existing faithfulness criteria for PRP are limited to coarse-grained LLM-based scoring without a clear definition or formulation.
This paper presents a pioneering exploration to quantify PRP faithfulness evaluation as a fine-grained and explainable criterion, which also serves as a reliable reference for faithfulness optimization.
Our criterion first discriminates persona statements into \emph{active} and \emph{passive} constraints by identifying the query-statement relevance.
Then, we incorporate all constraints following the principle that the AI character's response should be (a) entailed by active constraints and (b) not contradicted by passive constraints.
We translate this principle mathematically into a novel Active-Passive-Constraint (APC) score, a constraint-wise sum of statement-to-response natural language inference (NLI) scores weighted by constraint-query relevance scores.
In practice, we build the APC scoring system by symbolically distilling small NLI and relevance discriminators (300M parameters) from GPT-4 for efficiency, and both show high consistency with GPT-4's discrimination.
We validate the quality of the APC score against human evaluation based on example personas with tens of statements, and the results show a high correlation.
As the APC score could faithfully reflect the PRP quality, we further leverage it as a reward system in direct preference optimization (DPO) for better AI characters.
Our experiments offer a fine-grained and explainable comparison between existing PRP techniques, revealing their advantages and limitations.
We further find APC-based DPO to be one of the most competitive techniques for sticking with all constraints and can be well incorporated with other techniques.
We then extend the scale of the experiments to real persons with hundreds of statements and reach a consistent conclusion.
Finally, we provide comprehensive analyses and case studies to support the effectiveness of APC and APC-based DPO. Letian Peng, Jingbo Shang |
NeurIPS | 2 |
| 2024 | How Few Davids Improve One Goliath: Federated Learning in Resource-Skewed Edge Computing EnvironmentsabstractReal-world deployment of federated learning requires orchestrating clients with widely varied compute resources, from strong enterprise-grade devices in data centers to weak mobile and Web-of-Things devices. Prior works have attempted to downscale large models for weak devices and aggregate shared parts among heterogeneous models. A typical architectural assumption is that there are equally many strong and weak devices. In reality, however, we often encounter resource skew where a few (1 or 2) strong devices hold substantial data resources, alongside many weak devices. This poses challenges-the unshared portion of the large model rarely receives updates or gains benefits from weak collaborators. Jiayun Zhang, Shuheng Li, Haiyu Huang 0003, Zihan Wang 0001, Xiaohan Fu, Dezhi Hong, Rajesh K. Gupta 0001, Jingbo Shang |
WWW | 8 |
| 2023 | PrimeNet: Pre-training for Irregular Multivariate Time SeriesabstractReal-world applications often involve irregular time series, for which the time intervals between successive observations are non-uniform. Irregularity across multiple features in a multi-variate time series further results in a different subset of features at any given time (i.e., asynchronicity). Existing pre-training schemes for time-series, however, often assume regularity of time series and make no special treatment of irregularity. We argue that such irregularity offers insight about domain property of the data—for example, frequency of hospital visits may signal patient health condition—that can guide representation learning. In this work, we propose PrimeNet to learn a self-supervised representation for irregular multivariate time-series. Specifically, we design a time sensitive contrastive learning and data reconstruction task to pre-train a model. Irregular time-series exhibits considerable variations in sampling density over time. Hence, our triplet generation strategy follows the density of the original data points, preserving its native irregularity. Moreover, the sampling density variation over time makes data reconstruction difficult for different regions. Therefore, we design a data masking technique that always masks a constant time duration to accommodate reconstruction for regions of different sampling density. We learn with these tasks using unlabeled data to build a pre-trained model and fine-tune on a downstream task with limited labeled data, in contrast with existing fully supervised approach for irregular time-series, requiring large amounts of labeled data. Experiment results show that PrimeNet significantly outperforms state-of-the-art methods on naturally irregular and asynchronous data from Healthcare and IoT applications for several downstream tasks, including classification, interpolation, and regression. Ranak Roy Chowdhury, Jiacheng Li 0003, Xiyuan Zhang 0001, Dezhi Hong, Rajesh K. Gupta 0001, Jingbo Shang |
AAAI | 6 |
| 2023 | Towards Open-World Product Attribute Mining: A Lightly-Supervised ApproachabstractWe present a new task setting for attribute mining on e-commerce products, serving as a practical solution to extract open-world attributes without extensive human intervention.Our supervision comes from a high-quality seed attribute set bootstrapped from existing resources, and we aim to expand the attribute vocabulary of existing seed types, and also to discover any new attribute types automatically.A new dataset is created to support our setting, and our approach Amacer is proposed specifically to tackle the limited supervision.Especially, given that no direct supervision is available for those unseen new attributes, our novel formulation exploits selfsupervised heuristic and unsupervised latent attributes, which attains implicit semantic signals as additional supervision by leveraging product context.Experiments suggest that our approach surpasses various baselines by 12 F1, expanding attributes of existing types significantly by up to 12 times, and discovering values from 39% new types.Our data and code can be found at https://github.com/lxucs/woam. Liyan Xu, Jingbo Shang, Jinho D. Choi |
ACL (1) | 4 |
| 2023 | Unleashing the Power of Shared Label Structures for Human Activity RecognitionabstractCurrent human activity recognition (HAR) techniques regard activity labels as integer class IDs without explicitly modeling the semantics of class labels. We observe that different activity names often have shared structures. For example, "open door" and "open fridge" both have "open" as the action; "kicking soccer ball" and "playing tennis ball" both have "ball" as the object. Such shared structures in label names can be translated to the similarity in sensory data and modeling common structures would help uncover knowledge across different activities, especially for activities with limited samples. In this paper, we propose SHARE, a HAR framework that takes into account shared structures of label names for different activities. To exploit the shared structures, SHARE comprises an encoder for extracting features from input sensory time series and a decoder for generating label names as a token sequence. We also propose three label augmentation techniques to help the model more effectively capture semantic structures across activities, including a basic token-level augmentation, and two enhanced embedding-level and sequence-level augmentations utilizing the capabilities of pre-trained models. SHARE outperforms state-of-the-art HAR models in extensive experiments on seven HAR benchmark datasets. We also evaluate in few-shot learning and label imbalance settings and observe even more significant performance gap. Xiyuan Zhang 0001, Ranak Roy Chowdhury, Jiayun Zhang, Dezhi Hong, Rajesh K. Gupta 0001, Jingbo Shang |
CIKM | 6 |
| 2023 | WOT-Class: Weakly Supervised Open-world Text ClassificationabstractState-of-the-art weakly supervised text classification methods, while significantly reduced the required human supervision, still requires the supervision to cover all the classes of interest. This is never easy to meet in practice when human explore new, large corpora without complete pictures. In this paper, we work on a novel yet important problem of weakly supervised open-world text classification, where supervision is only needed for a few examples from a few known classes and the machine should handle both known and unknown classes in test time. General open-world classification has been studied mostly using image classification; however, existing methods typically assume the availability of sufficient known-class supervision and strong unknown-class prior knowledge (e.g., the number and/or data distribution). We propose a novel framework øur that lifts those strong assumptions. Specifically, it follows an iterative process of (a) clustering text to new classes, (b) mining and ranking indicative words for each class, and (c) merging redundant classes by using the overlapped indicative words as a bridge. Extensive experiments on 7 popular text classification datasets demonstrate that øur outperforms strong baselines consistently with a large margin, attaining 23.33% greater average absolute macro-F1 over existing approaches across all datasets. Such competent accuracy illuminates the practical potential of further reducing human effort for text classification. Tianle Wang 0003, Zihan Wang 0001, Weitang Liu, Jingbo Shang |
CIKM | 4 |
| 2023 | ClusterLLM: Large Language Models as a Guide for Text ClusteringabstractWe introduce CLUSTERLLM, a novel text clustering framework that leverages feedback from an instruction-tuned large language model, such as ChatGPT.Compared with traditional unsupervised methods that builds upon "small" embedders, CLUSTERLLM exhibits two intriguing advantages: (1) it enjoys the emergent capability of LLM even if its embeddings are inaccessible; and (2) it understands the user's preference on clustering through textual instruction and/or a few annotated data.First, we prompt ChatGPT for insights on clustering perspective by constructing hard triplet questions , where A, B and C are similar data points that belong to different clusters according to small embedder.We empirically show that this strategy is both effective for fine-tuning small embedder and cost-efficient to query ChatGPT.Second, we prompt ChatGPT for helps on clustering granularity by carefully designed pairwise questions , and tune the granularity from cluster hierarchies that is the most consistent with the ChatGPT answers.Extensive experiments on 14 datasets show that CLUSTERLLM consistently improves clustering quality, at an average cost of ∼$0.6 1 per dataset.The code will be available at https: //github.com/zhang-yu-wei/ClusterLLM. Yuwei Zhang 0001, Zihan Wang 0001, Jingbo Shang |
EMNLP | 3 |
| 2023 | Debiasing Made State-of-the-art: Revisiting the Simple Seed-based Weak Supervision for Text ClassificationabstractRecent advances in weakly supervised text classification mostly focus on designing sophisticated methods to turn high-level human heuristics into quality pseudo-labels.In this paper, we revisit the seed matching-based method, which is arguably the simplest way to generate pseudo-labels, and show that its power was greatly underestimated.We show that the limited performance of seed matching is largely due to the label bias injected by the simple seed-match rule, which prevents the classifier from learning reliable confidence for selecting high-quality pseudo-labels.Interestingly, simply deleting the seed words present in the matched input texts can mitigate the label bias and help learn better confidence.Subsequently, the performance achieved by seed matching can be improved significantly, making it on par with or even better than the state-of-theart.Furthermore, to handle the case when the seed words are not made known, we propose to simply delete the word tokens in the input text randomly with a high deletion ratio.Remarkably, seed matching equipped with this random deletion method can often achieve even better performance than that with seed deletion.We refer to our method as SimSeed, which is publicly available 1 . Chengyu Dong, Zihan Wang 0001, Jingbo Shang |
EMNLP | 3 |
| 2023 | Open-world Semi-supervised Generalized Relation Discovery Aligned in a Real-world SettingabstractOpen-world Relation Extraction (OpenRE) has recently garnered significant attention.However, existing approaches tend to oversimplify the problem by assuming that all instances of unlabeled data belong to novel classes, thereby limiting the practicality of these methods.We argue that the OpenRE setting should be more aligned with the characteristics of real-world data.Specifically, we propose two key improvements: (a) unlabeled data should encompass known and novel classes, including negative instances; and (b) the set of novel classes should represent long-tail relation types.Furthermore, we observe that popular relations can often be implicitly inferred through specific patterns, while long-tail relations tend to be explicitly expressed.Motivated by these insights, we present a method called KNoRD (Known and Novel Relation Discovery), which effectively classifies explicitly and implicitly expressed relations from known and novel classes within unlabeled data.Experimental evaluations on several Open-world RE benchmarks demonstrate that KNoRD consistently outperforms existing methods, achieving significant gains. William Hogan, Jiacheng Li 0003, Jingbo Shang |
EMNLP | 3 |
| 2023 | Goal-Driven Explainable Clustering via Language DescriptionsabstractUnsupervised clustering is widely used to explore large corpora, but existing formulations neither consider the users' goals nor explain clusters' meanings.We propose a new task formulation, "Goal-Driven Clustering with Explanations" (GOALEX), which represents both the goal and the explanations as free-form language descriptions.For example, to categorize the errors made by a summarization system, the input to GOALEX is a corpus of annotatorwritten comments for system-generated summaries and a goal "cluster the comments based on why the annotators think the summary is imperfect.";the outputs are text clusters each with an explanation ("this cluster mentions that the summary misses important context information."),which relates to the goal and accurately explains which comments should (not) belong to a cluster.To tackle GOALEX, we prompt a language model with "[corpus subset] + [goal] + Brainstorm a list of explanations each representing a cluster.";then we classify whether each sample belongs to a cluster based on its explanation; finally, we use integer linear programming to select a subset of candidate clusters to cover most samples while minimizing overlaps.Under both automatic and human evaluation on corpora with or without labels, our method produces more accurate and goalrelated explanations than prior methods. Zihan Wang 0001, Jingbo Shang, Ruiqi Zhong |
EMNLP | 2 |
| 2023 | Towards Diverse and Coherent Augmentation for Time-Series ForecastingabstractTime-series data augmentation mitigates the issue of insufficient training data for deep learning models. Yet, existing augmentation methods are mainly designed for classification, where class labels can be preserved even if augmentation alters the temporal dynamics. We note that augmentation designed for forecasting requires diversity as well as coherence with the original temporal dynamics. As time-series data generated by real-life physical processes exhibit characteristics in both the time and frequency domains, we propose to combine Spectral and Time Augmentation (STAug) for generating more diverse and coherent samples. Specifically, in the frequency domain, we use the Empirical Mode Decomposition to decompose a time series and reassemble the subcomponents with random weights. This way, we generate diverse samples while being coherent with the original temporal relationships as they contain the same set of base components. In the time domain, we adapt a mix-up strategy that generates diverse as well as linearly in-between coherent samples. Experiments on five real-world time-series datasets demonstrate that STAug outperforms the base models without data augmentation as well as state-of-the-art augmentation methods. Xiyuan Zhang 0001, Ranak Roy Chowdhury, Jingbo Shang, Rajesh K. Gupta 0001, Dezhi Hong |
ICASSP | 3 |
| 2023 | Learning Concise and Descriptive Attributes for Visual RecognitionabstractRecent advances in foundation models present new opportunities for interpretable visual recognition – one can first query Large Language Models (LLMs) to obtain a set of attributes that describe each class, then apply vision-language models to classify images via these attributes. Pioneering work shows that querying thousands of attributes can achieve performance competitive with image features. However, our further investigation on 8 datasets reveals that LLM-generated attributes in a large quantity perform almost the same as random words. This surprising finding suggests that significant noise may be present in these attributes. We hypothesize that there exist subsets of attributes that can maintain the classification performance with much smaller sizes, and propose a novel learning-to-search method to discover those concise sets of attributes. As a result, on the CUB dataset, our method achieves performance close to that of massive LLM-generated attributes (e.g., 10k attributes for CUB), yet using only 32 attributes in total to distinguish 200 bird species. Furthermore, our new paradigm demonstrates several additional benefits: higher interpretability and interactivity for humans, and the ability to summarize knowledge for a recognition task. An Yan 0003, Yu Wang 0170, Yiwu Zhong, Chengyu Dong, Zexue He, William Yang Wang, Jingbo Shang, Julian J. McAuley |
ICCV | 8 |
| 2023 | On Compositional Uncertainty Quantification for Seq2seq Graph Parsing
Zi Lin, Du Phan, Panupong Pasupat, Jeremiah Z. Liu, Jingbo Shang |
ICLR | 5 |
| 2023 | Understand and Modularize Generator Optimization in ELECTRA-style PretrainingabstractDespite the effectiveness of ELECTRA-style pre-training, their performance is dependent on the careful selection of the model size for the auxiliary generator, leading to high trial-and-error costs. In this paper, we present the first systematic study of this problem. Our theoretical investigation highlights the importance of controlling the generator capacity in ELECTRA-style training. Meanwhile, we found it is not handled properly in the original ELECTRA design, leading to the sensitivity issue. Specifically, since adaptive optimizers like Adam will cripple the weighing of individual losses in the joint optimization, the original design fails to control the generator training effectively. To regain control over the generator, we modularize the generator optimization by decoupling the generator optimizer and discriminator optimizer completely, instead of simply relying on the weighted objective combination. Our simple technique reduced the sensitivity of ELECTRA training significantly and obtains considerable performance gain compared to the original design. Chengyu Dong, Hao Cheng 0002, Jingbo Shang, Jianfeng Gao 0001, Xiaodong Liu 0003 |
ICML | 4 |
| 2023 | Gradient-based Wang-Landau Algorithm: A Novel Sampler for Output Distribution of Neural Networks over the Input SpaceabstractThe output distribution of a neural network (NN) over the entire input space captures the complete input-output mapping relationship, offering in- sights toward a more comprehensive NN under- standing. Exhaustive enumeration or traditional Monte Carlo methods for the entire input space can exhibit impractical sampling time, especially for high-dimensional inputs. To make such difficult sampling computationally feasible, in this paper, we propose a novel Gradient-based Wang-Landau (GWL) sampler. We first draw the connection between the output distribution of a NN and the density of states (DOS) of a physical system. Then, we renovate the classic sampler for the DOS problem, Wang-Landau algorithm, by re-placing its random proposals with gradient-based Monte Carlo proposals. This way, our GWL sampler investigates the under-explored subsets of the input space much more efficiently. Extensive experiments have verified the accuracy of the output distribution generated by GWL and also showcased several interesting findings - for example, in a binary image classification task, both CNN and ResNet mapped the majority of human unrecognizable images to very negative logit values. Weitang Liu, Yi-Zhuang You, Ying-Wai Li, Jingbo Shang |
ICML | 4 |
| 2023 | Minimally Supervised Contextual Inference from Human Mobility: An Iterative Collaborative Distillation FrameworkabstractThe context about trips and users from mobility data is valuable for mobile service providers to understand their customers and improve their services. Existing inference methods require a large number of labels for training, which is hard to meet in practice. In this paper, we study a more practical yet challenging setting—contextual inference using mobility data with minimal supervision (i.e., a few labels per class and massive unlabeled data). A typical solution is to apply semi-supervised methods that follow a self-training framework to bootstrap a model based on all features. However, using a limited labeled set brings high risk of overfitting to self-training, leading to unsatisfactory performance. We propose a novel collaborative distillation framework STCOLAB. It sequentially trains spatial and temporal modules at each iteration following the supervision of ground-truth labels. In addition, it distills knowledge to the module being trained using the logits produced by the latest trained module of the other modality, thereby mutually calibrating the two modules and combining the knowledge from both modalities. Extensive experiments on two real-world datasets show STCOLAB achieves significantly more accurate contextual inference than various baselines. Jiayun Zhang, Xinyang Zhang 0002, Dezhi Hong, Rajesh K. Gupta 0001, Jingbo Shang |
IJCAI | 5 |
| 2023 | UCEpic: Unifying Aspect Planning and Lexical Constraints for Generating Explanations in RecommendationabstractPersonalized natural language generation for explainable recommendations plays a key role in justifying why a recommendation might match a user's interests. Existing models usually control the generation process by aspect planning. While promising, these aspect-planning methods struggle to generate specific information correctly, which prevents generated explanations from being convincing. In this paper, we claim that introducing lexical constraints can alleviate the above issues. We propose a model, UCEpic, that generates high-quality personalized explanations for recommendation results by unifying aspect planning and lexical constraints in an insertion-based generation manner. Jiacheng Li 0003, Zhankui He, Jingbo Shang, Julian J. McAuley |
KDD | 3 |
| 2023 | Text Is All You Need: Learning Language Representations for Sequential RecommendationabstractSequential recommendation aims to model dynamic user behavior from historical interactions. Existing methods rely on either explicit item IDs or general textual features for sequence modeling to understand user preferences. While promising, these approaches still struggle to model cold-start items or transfer knowledge to new datasets. In this paper, we propose to model user preferences and item features as language representations that can be generalized to new items and datasets. To this end, we present a novel framework, named Recformer, which effectively learns language representations for sequential recommendation. Specifically, we propose to formulate an item as a "sentence" (word sequence) by flattening item key-value attributes described by text so that an item sequence for a user becomes a sequence of sentences. For recommendation, Recformer is trained to understand the "sentence" sequence and retrieve the next "sentence". To encode item sequences, we design a bi-directional Transformer similar to the model Longformer but with different embedding layers for sequential recommendation. For effective representation learning, we propose novel pretraining and finetuning methods which combine language understanding and recommendation tasks. Therefore, Recformer can effectively recommend the next item based on language representations. Extensive experiments conducted on six datasets demonstrate the effectiveness of Recformer for sequential recommendation, especially in low-resource and cold-start settings. Jiacheng Li 0003, Jin Li 0003, Jinmiao Fu, Jingbo Shang, Julian J. McAuley |
KDD | 6 |
| 2023 | Navigating Alignment for Non-identical Client Class Sets: A Label Name-Anchored Federated Learning FrameworkabstractTraditional federated classification methods, even those designed for non-IID clients, assume that each client annotates its local data with respect to the same universal class set. In this paper, we focus on a more general yet practical setting, non-identical client class sets, where clients focus on their own (different or even non-overlapping) class sets and seek a global model that works for the union of these classes. If one views classification as finding the best match between representations produced by data/label encoder, such heterogeneity in client class sets poses a new significant challenge-local encoders at different clients may operate in different and even independent latent spaces, making it hard to aggregate at the server. We propose a novel framework, FedAlign1, to align the latent spaces across clients from both label and data perspectives. From a label perspective, we leverage the expressive natural language class names as a common ground for label encoders to anchor class representations and guide the data encoder learning across clients. From a data perspective, during local training, we regard the global class representations as anchors and leverage the data points that are close/far enough to the anchors of locally-unaware classes to align the data encoders across clients. Our theoretical analysis of the generalization performance and extensive experiments on four real-world datasets of different tasks confirm that FedAlign outperforms various state-of-the-art (non-IID) federated classification methods. Jiayun Zhang, Xiyuan Zhang 0001, Xinyang Zhang 0002, Dezhi Hong, Rajesh K. Gupta 0001, Jingbo Shang |
KDD | 6 |
| 2023 | Physics-Informed Data Denoising for Real-Life Sensing SystemsabstractSensors measuring real-life physical processes are ubiquitous in today's interconnected world. These sensors inherently bear noise that often adversely affects the performance and reliability of the systems they support. Classic filtering approaches introduce strong assumption on the time or frequency characteristics of sensory measurements, while learning-based denoising approaches typically rely on using ground truth clean data to train a denoising model, which is often challenging or prohibitive to obtain for many real-world applications. We observe that in many scenarios, the relationships between different sensor measurements (e.g., location and acceleration) are analytically described by laws of physics (e.g., second-order differential equation). By incorporating such physics constraints, we can guide the denoising process to improve performance even in the absence of ground truth data. In light of this, we design a physics-informed denoising model that leverages the inherent algebraic relationships between different measurements governed by the underlying physics. By obviating the need for ground truth clean data, our method offers a practical denoising solution for real-world applications. We conducted experiments in various domains, including inertial navigation, CO2 monitoring, and HVAC control, and achieved state-of-the-art performance compared with existing denoising methods. Our method can denoise data in real time (4ms for a sequence of 1s) for low-cost noisy sensors and produces results that closely align with those from high-precision, high-cost alternatives, leading to an efficient, cost-effective approach for more accurate sensor-based systems. Xiyuan Zhang 0001, Xiaohan Fu, Diyan Teng, Chengyu Dong, Keerthivasan Vijayakumar, Jiayun Zhang, Ranak Roy Chowdhury, Junsheng Han, Dezhi Hong, Rashmi Kulkarni, Jingbo Shang, Rajesh K. Gupta 0001 |
SenSys | 11 |
| 2022 | Unsupervised Deep Keyphrase GenerationabstractKeyphrase generation aims to summarize long documents with a collection of salient phrases. Deep neural models have demonstrated remarkable success in this task, with the capability of predicting keyphrases that are even absent from a document. However, such abstractiveness is acquired at the expense of a substantial amount of annotated data. In this paper, we present a novel method for keyphrase generation, AutoKeyGen, without the supervision of any annotated doc-keyphrase pairs. Motivated by the observation that an absent keyphrase in a document may appear in other places, in whole or in part, we construct a phrase bank by pooling all phrases extracted from a corpus. With this phrase bank, we assign phrase candidates to new documents by a simple partial matching algorithm, and then we rank these candidates by their relevance to the document from both lexical and semantic perspectives. Moreover, we bootstrap a deep generative model using these top-ranked pseudo keyphrases to produce more absent candidates. Extensive experiments demonstrate that AutoKeyGen outperforms all unsupervised baselines and can even beat a strong supervised method in certain cases. Xianjie Shen, Yinghan Wang, Jingbo Shang |
AAAI | 4 |
| 2022 | Towards Comprehensive Patent Approval Predictions: Beyond Traditional Document ClassificationabstractXiaochen Gao, Zhaoyi Hou, Yifei Ning, Kewen Zhao, Beilei He, Jingbo Shang, Vish Krishnan. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022. Xiaochen Gao, Zhaoyi Hou, Yifei Ning, Kewen Zhao, Beilei He, Jingbo Shang, Vish Krishnan |
ACL (1) | 6 |
| 2022 | Phrase-aware Unsupervised Constituency ParsingabstractRecent studies have achieved inspiring success in unsupervised grammar induction using masked language modeling (MLM) as the proxy task.Despite their high accuracy in identifying low-level structures, prior arts tend to struggle in capturing high-level structures like clauses, since the MLM task usually only requires information from local context.In this work, we revisit LM-based constituency parsing from a phrase-centered perspective.Inspired by the natural reading process of human readers, we propose to regularize the parser with phrases extracted by an unsupervised phrase tagger to help the LM model quickly manage low-level structures.For a better understanding of high-level structures, we propose a phrase-guided masking strategy for LM to emphasize more on reconstructing nonphrase words.We show that the initial phrase regularization serves as an effective bootstrap, and phrase-guided masking improves the identification of high-level structures.Experiments on the public benchmark with two different backbone models demonstrate the effectiveness and generality of our method. Xiaotao Gu, Yikang Shen, Jingbo Shang, Jiawei Han 0001 |
ACL (1) | 4 |
| 2022 | UCTopic: Unsupervised Contrastive Learning for Phrase Representations and Topic MiningabstractHigh-quality phrase representations are essential to finding topics and related terms in documents (a.k.a.topic mining).Existing phrase representation learning methods either simply combine unigram representations in a contextfree manner or rely on extensive annotations to learn context-aware knowledge.In this paper, we propose UCTOPIC, a novel unsupervised contrastive learning framework for context-aware phrase representations and topic mining.UCTOPIC is pretrained in a large scale to distinguish if the contexts of two phrase mentions have the same semantics.The key to pretraining is positive pair construction from our phrase-oriented assumptions.However, we find traditional in-batch negatives cause performance decay when finetuning on a dataset with small topic numbers.Hence, we propose cluster-assisted contrastive learning (CCL) which largely reduces noisy negatives by selecting negatives from clusters and further improves phrase representations for topics accordingly.UCTOPIC outperforms the state-of-the-art phrase representation model by 38.2% NMI in average on four entity clustering tasks.Comprehensive evaluation on topic mining shows that UCTOPIC can extract coherent and diverse topical phrases. Jiacheng Li 0003, Jingbo Shang, Julian J. McAuley |
ACL (1) | 2 |
| 2022 | MGDoc: Pre-training with Multi-granular Hierarchy for Document Image UnderstandingabstractZilong Wang, Jiuxiang Gu, Chris Tensmeyer, Nikolaos Barmpalios, Ani Nenkova, Tong Sun, Jingbo Shang, Vlad Morariu. Proceedings of the 2022 Conference on Empirical Methods in Natural Language Processing. 2022. Zilong Wang 0002, Jiuxiang Gu, Chris Tensmeyer, Nikolaos Barmpalios, Ani Nenkova, Tong Sun 0005, Jingbo Shang, Vlad I. Morariu |
EMNLP | 7 |
| 2022 | Fine-grained Contrastive Learning for Relation ExtractionabstractRecent relation extraction (RE) works have shown encouraging improvements by conducting contrastive learning on silver labels generated by distant supervision before fine-tuning on gold labels.Existing methods typically assume all these silver labels are accurate and treat them equally; however, distant supervision is inevitably noisy-some silver labels are more reliable than others.In this paper, we propose fine-grained contrastive learning (FineCL) for RE, which leverages fine-grained information about which silver labels are and are not noisy to improve the quality of learned relationship representations for RE.We first assess the quality of silver labels via a simple and automatic approach we call "learning order denoising," where we train a language model to learn these relations and record the order of learned training instances.We show that learning order largely corresponds to label accuracy-early-learned silver labels have, on average, more accurate labels than later-learned silver labels.Then, during pre-training, we increase the weights of accurate labels within a novel contrastive learning objective.Experiments on several RE benchmarks show that FineCL makes consistent and significant performance gains over state-of-the-art methods. William Hogan, Jiacheng Li 0003, Jingbo Shang |
EMNLP | 3 |
| 2022 | WeDef: Weakly Supervised Backdoor Defense for Text ClassificationabstractExisting backdoor defense methods are only effective for limited trigger types.To defend different trigger types at once, we start from the class-irrelevant nature of the poisoning process and propose a novel weakly supervised backdoor defense framework WeDef.Recent advances in weak supervision make it possible to train a reasonably accurate text classifier using only a small number of user-provided, class-indicative seed words.Such seed words shall be considered independent of the triggers.Therefore, a weakly supervised text classifier trained by only the poisoned documents without their labels will likely have no backdoor.Inspired by this observation, in WeDef, we define the reliability of samples based on whether the predictions of the weak classifier agree with their labels in the poisoned training set.We further improve the results through a two-phase sanitization: (1) iteratively refine the weak classifier based on the reliable samples and (2) train a binary poison classifier by distinguishing the most unreliable samples from the most reliable samples.Finally, we train the sanitized model on the samples that the poison classifier predicts as benign.Extensive experiments show that WeDef is effective against popular trigger-based attacks (e.g., words, sentences, and paraphrases), outperforming existing defense methods. Lesheng Jin, Zihan Wang 0001, Jingbo Shang |
EMNLP | 3 |
| 2022 | Neural-Symbolic Inference for Robust Autoregressive Graph Parsing via Compositional Uncertainty QuantificationabstractPre-trained seq2seq models excel at graph semantic parsing with rich annotated data, but generalize worse to out-of-distribution (OOD) and long-tail examples.In comparison, symbolic parsers under-perform on populationlevel metrics, but exhibit unique strength in OOD and tail generalization.In this work, we study compositionality-aware approach to neural-symbolic inference informed by model confidence, performing fine-grained neuralsymbolic reasoning at subgraph level (i.e., nodes and edges) and precisely targeting subgraph components with high uncertainty in the neural parser.As a result, the method combines the distinct strength of the neural and symbolic approaches in capturing different aspects of the graph prediction, leading to well-rounded generalization performance both across domains and in the tail.We empirically investigate the approach in the English Resource Grammar (ERG) parsing problem on a diverse suite of standard in-domain and seven OOD corpora.Our approach leads to 35.26% and 35.60% error reduction in aggregated SMATCH score over neural and symbolic approaches respectively, and 14% absolute accuracy gain in key tail linguistic categories over the neural model, outperforming prior state-of-art methods that do not account for compositionality or uncertainty. Zi Lin, Jeremiah Z. Liu, Jingbo Shang |
EMNLP | 3 |
| 2022 | Leveraging QA Datasets to Improve Generative Data AugmentationabstractThe ability of generative language models (GLMs) to generate text has improved considerably in the last few years, enabling their use for generative data augmentation.In this work, we propose CONDA, an approach to further improve GLMs' ability to generate synthetic data by reformulating data generation as context generation for a given question-answer (QA) pair and leveraging QA datasets for training context generators.Then, we cast downstream tasks into the same question answering format and adapt the fine-tuned context generators to the target task domain.Finally, we use the fine-tuned GLM to generate relevant contexts, which are in turn used as synthetic training data for their corresponding tasks.We perform extensive experiments on multiple classification datasets and demonstrate substantial improvements in performance for both few-and zeroshot settings.Our analysis reveals that QA datasets that require high-level reasoning abilities (e.g., abstractive and common-sense QA datasets) tend to give the best boost in performance in both few-shot and zero-shot settings. Dheeraj Mekala, Tu Vu, Timo Schick, Jingbo Shang |
EMNLP | 4 |
| 2022 | TARNet: Task-Aware Reconstruction for Time-Series TransformerabstractTime-series data contains temporal order information that can guide representation learning for predictive end tasks (e.g., classification, regression). Recently, there are some attempts to leverage such order information to first pre-train time-series models by reconstructing time-series values of randomly masked time segments, followed by an end-task fine-tuning on the same dataset, demonstrating improved end-task performance. However, this learning paradigm decouples data reconstruction from the end task. We argue that the representations learnt in this way are not informed by the end task and may, therefore, be sub-optimal for the end-task performance. In fact, the importance of different timestamps can vary significantly in different end tasks. We believe that representations learnt by reconstructing important timestamps would be a better strategy for improving end-task performance. In this work, we propose TARNet, Task-Aware Reconstruction Network, a new model using Transformers to learn task-aware data reconstruction that augments end-task performance. Specifically, we design a data-driven masking strategy that uses self-attention score distribution from end-task training to sample timestamps deemed important by the end task. Then, we mask out data at those timestamps and reconstruct them, thereby making the reconstruction task-aware. This reconstruction task is trained alternately with the end task at every epoch, sharing parameters in a single model, allowing the representation learnt through reconstruction to improve end-task performance. Extensive experiments on tens of classification and regression datasets show that TARNet significantly outperforms state-of-the-art baseline models across all evaluation metrics. Ranak Roy Chowdhury, Xiyuan Zhang 0001, Jingbo Shang, Rajesh K. Gupta 0001, Dezhi Hong |
KDD | 3 |
| 2022 | Label Noise in Adversarial Training: A Novel Perspective to Study Robust OverfittingabstractWe show that label noise exists in adversarial training. Such label noise is due to the mismatch between the true label distribution of adversarial examples and the label inherited from clean examples – the true label distribution is distorted by the adversarial perturbation, but is neglected by the common practice that inherits labels from clean examples. Recognizing label noise sheds insights on the prevalence of robust overfitting in adversarial training, and explains its intriguing dependence on perturbation radius and data quality. Also, our label noise perspective aligns well with our observations of the epoch-wise double descent in adversarial training. Guided by our analyses, we proposed a method to automatically calibrate the label to address the label noise and robust overfitting. Our method achieves consistent performance improvements across various models and datasets without introducing new hyper-parameters or additional tuning. Chengyu Dong, Jingbo Shang |
NeurIPS | 3 |
| 2022 | SQEE: A Machine Perception Approach to Sensing Quality Evaluation at the Edge by Uncertainty QuantificationabstractCyber-physical systems are starting to adopt neural network (NN) models for a variety of smart sensing applications. While several efforts seek better NN architectures for system performance improvement, few attempts have been made to study the deployment of these systems in the field. Proper deployment of these systems is critical to achieving ideal performance, but the current practice is largely empirical via trials and errors, lacking a measure of quality. Sensing quality should reflect the impact on the performance of NN models that drive machine perception tasks. However, traditional approaches either evaluate statistical difference that exists objectively, or model the quality subjectively via human perception. Shuheng Li, Jingbo Shang, Rajesh K. Gupta 0001, Dezhi Hong |
SenSys | 2 |
| 2022 | ESC-GAN: Extending Spatial Coverage of Physical SensorsabstractScientific discoveries and studies about our physical world have long benefited from large-scale and planetary sensing, from weather forecasting to wildfire monitoring. However, the limited deployment of sensors in the environment due to cost or physical access constraints has lagged behind our ever-growing need for increased data coverage and higher resolution, impeding timely and precise monitoring and understanding of the environment. Therefore, we seek to extend the spatial coverage of analysis based on existing sensory data, that is, to "generate" data for locations where no historical data exists. This problem is fundamentally different and more challenging than the traditional spatio-temporal imputation that assumes data for any particular location are only partially missing across time. Inspired by the success of Generative Adversarial Network (GAN) in imputation, we propose a novel ESC-GAN. We observe that there are local patterns in nearby locations, as well as trends in a global manner (e.g., temperature drops as altitude increases regardless of the location). As local patterns may exhibit at different scales (from meters to kilometers), we employ a multi-branch generator to aggregate information of different granularity. More specifically, each branch in the generator contains 1) randomly masked 3D partial convolutions at different resolutions to capture the local patterns and 2) global attention modules for global similarity. Next, we adversarially train a 3D convolution-based discriminator to distinguish the generator's output from the ground truth. Extensive experiments on three geo-sensor datasets demonstrate that ESC-GAN outperforms state-of-the-art methods on extending spatial coverage and also achieves the best results on a traditional spatio-temporal imputation task. Xiyuan Zhang 0001, Ranak Roy Chowdhury, Jingbo Shang, Rajesh K. Gupta 0001, Dezhi Hong |
WSDM | 3 |
| 2022 | OA-Mine: Open-World Attribute Mining for E-Commerce Products with Weak SupervisionabstractAutomatic extraction of product attributes from their textual descriptions is essential for online shopper experience. One inherent challenge of this task is the emerging nature of e-commerce products — we see new types of products with their unique set of new attributes constantly. Most prior works on this matter mine new values for a set of known attributes but cannot handle new attributes that arose from constantly changing data. In this work, we study the attribute mining problem in an open-world setting to extract novel attributes and their values. Instead of providing comprehensive training data, the user only needs to provide a few examples for a few known attribute types as weak supervision. We propose a principled framework that first generates attribute value candidates and then groups them into clusters of attributes. The candidate generation step probes a pre-trained language model to extract phrases from product titles. Then, an attribute-aware fine-tuning method optimizes a multitask objective and shapes the language model representation to be attribute-discriminative. Finally, we discover new attributes and values through the self-ensemble of our framework, which handles the open-world challenge. We run extensive experiments on a large distantly annotated development set and a gold standard human-annotated test set that we collected. Our model significantly outperforms strong baselines and can generalize to unseen attributes and product types. Xinyang Zhang 0002, Xin Dong 0001, Jingbo Shang, Christos Faloutsos, Jiawei Han 0001 |
WWW | 5 |
| 2021 | Weakly Supervised Named Entity Tagging with Learnable Logical RulesabstractJiacheng Li, Haibo Ding, Jingbo Shang, Julian McAuley, Zhe Feng. Proceedings of the 59th Annual Meeting of the Association for Computational Linguistics and the 11th International Joint Conference on Natural Language Processing (Volume 1: Long Papers). 2021. Jiacheng Li 0003, Haibo Ding, Jingbo Shang, Julian J. McAuley, Zhe Feng 0003 |
ACL/IJCNLP (1) | 3 |
| 2021 | Coarse2Fine: Fine-grained Text Classification on Coarsely-grained Annotated DataabstractExisting text classification methods mainly focus on a fixed label set, whereas many realworld applications require extending to new fine-grained classes as the number of samples per label increases.To accommodate such requirements, we introduce a new problem called coarse-to-fine grained classification, which aims to perform fine-grained classification on coarsely annotated data.Instead of asking for new fine-grained human annotations, we opt to leverage label surface names as the only human guidance and weave in rich pretrained generative language models into the iterative weak supervision strategy.Specifically, we first propose a label-conditioned finetuning formulation to attune these generators for our task.Furthermore, we devise a regularization objective based on the coarse-fine label constraints derived from our problem setting, giving us even further improvements over the prior formulation.Our framework uses the fine-tuned generative models to sample pseudo-training data for training the classifier, and bootstraps on real unlabeled data for model refinement.Extensive experiments and case studies on two real-world datasets demonstrate superior performance over SOTA zeroshot classification baselines. Dheeraj Mekala, Varun Gangal, Jingbo Shang |
EMNLP (1) | 3 |
| 2021 | "Average" Approximates "First Principal Component"? An Empirical Analysis on Representations from Neural Language ModelsabstractContextualized representations based on neural language models have furthered the state of the art in various NLP tasks.Despite its great success, the nature of such representations remains a mystery.In this paper, we present an empirical property of these representations-"average" ≈ "first principal component".Specifically, experiments show that the average of these representations shares almost the same direction as the first principal component of the matrix whose columns are these representations.We believe this explains why the average representation is always a simple yet strong baseline.Our further examinations show that this property also holds in more challenging scenarios, for example, when the representations are from a model right after its random initialization.Therefore, we conjecture that this property is intrinsic to the distribution of representations and not necessarily related to the input structure.We realize that these representations empirically follow a normal distribution for each dimension, and by assuming this is true, we demonstrate that the empirical property can be in fact derived mathematically. Zihan Wang 0001, Chengyu Dong, Jingbo Shang |
EMNLP (1) | 3 |
| 2021 | LayoutReader: Pre-training of Text and Layout for Reading Order DetectionabstractReading order detection is the cornerstone to understanding visually-rich documents (e.g., receipts and forms).Unfortunately, no existing work took advantage of advanced deep learning models because it is too laborious to annotate a large enough dataset.We observe that the reading order of WORD documents is embedded in their XML metadata; meanwhile, it is easy to convert WORD documents to PDFs or images.Therefore, in an automated manner, we construct ReadingBank, a benchmark dataset that contains reading order, text, and layout information for 500,000 document images covering a wide spectrum of document types.This first-ever large-scale dataset unleashes the power of deep neural networks for reading order detection.Specifically, our proposed LayoutReader captures the text and layout information for reading order prediction using the seq2seq model.It performs almost perfectly in reading order detection and significantly improves both open-source and commercial OCR engines in ordering text lines in their results in our experiments.The dataset and models are publicly available at https: //aka.ms/layoutreader. Zilong Wang 0002, Yiheng Xu, Lei Cui 0001, Jingbo Shang, Furu Wei |
EMNLP (1) | 4 |
| 2021 | UCPhrase: Unsupervised Context-aware Quality Phrase TaggingabstractIdentifying and understanding quality phrases from context is a fundamental task in text mining. The most challenging part of this task arguably lies in uncommon, emerging, and domain-specific phrases. The infrequent nature of these phrases significantly hurts the performance of phrase mining methods that rely on sufficient phrase occurrences in the input corpus. Context-aware tagging models, though not restricted by frequency, heavily rely on domain experts for either massive sentence-level gold labels or handcrafted gazetteers. In this work, we propose UCPhrase, a novel unsupervised context-aware quality phrase tagger. Specifically, we induce high-quality phrase spans as silver labels from consistently co-occurring word sequences within each document. Compared with typical context-agnostic distant supervision based on existing knowledge bases (KBs), our silver labels root deeply in the input domain and context, thus having unique advantages in preserving contextual completeness and capturing emerging, out-of-KB phrases. Training a conventional neural tagger based on silver labels usually faces the risk of overfitting phrase surface names. Alternatively, we observe that the contextualized attention maps generated from a transformer-based neural language model effectively reveal the connections between words in a surface-agnostic way. Therefore, we pair such attention maps with the silver labels to train a lightweight span prediction model, which can be applied to new input to recognize (unseen) quality phrases regardless of their surface names or frequency. Thorough experiments on various tasks and datasets, including corpus-level phrase ranking, document-level keyphrase extraction, and sentence-level phrase tagging, demonstrate the superiority of our design over state-of-the-art pre-trained, unsupervised, and distantly supervised methods. Xiaotao Gu, Zihan Wang 0001, Zhenyu Bi, Yu Meng 0001, Jiawei Han 0001, Jingbo Shang |
KDD | 7 |
| 2021 | TaxoClass: Hierarchical Multi-Label Text Classification Using Only Class NamesabstractJiaming Shen, Wenda Qiu, Yu Meng, Jingbo Shang, Xiang Ren, Jiawei Han. Proceedings of the 2021 Conference of the North American Chapter of the Association for Computational Linguistics: Human Language Technologies. 2021. Wenda Qiu, Yu Meng 0001, Jingbo Shang, Xiang Ren 0001, Jiawei Han 0001 |
NAACL-HLT | 4 |
| 2021 | X-Class: Text Classification with Extremely Weak SupervisionabstractIn this paper, we explore text classification with extremely weak supervision, i.e., only relying on the surface text of class names.This is a more challenging setting than the seed-driven weak supervision, which allows a few seed words per class.We opt to attack this problem from a representation learning perspective-ideal document representations should lead to nearly the same results between clustering and the desired classification.In particular, one can classify the same corpus differently (e.g., based on topics and locations), so document representations should be adaptive to the given class names.We propose a novel framework X-Class to realize the adaptive representations.Specifically, we first estimate class representations by incrementally adding the most similar word to each class until inconsistency arises.Following a tailored mixture of class attention mechanisms, we obtain the document representation via a weighted average of contextualized word representations.With the prior of each document assigned to its nearest class, we then cluster and align the documents to classes.Finally, we pick the most confident documents from each cluster to train a text classifier.Extensive experiments demonstrate that X-Class can rival and even outperform seed-driven weakly supervised methods on 7 benchmark datasets. Zihan Wang 0001, Dheeraj Mekala, Jingbo Shang |
NAACL-HLT | 3 |
| 2021 | "Misc"-Aware Weakly Supervised Aspect ClassificationabstractAspect classification, identifying aspects of text segments, facilitates numerous applications, such as sentiment analysis and review summarization.To alleviate the extensive human effort required by existing aspect classification methods, in this paper, we focus on a weakly supervised setting-the model input only contains domainspecific raw texts and a few seed words per pre-defined aspect.We identify a unique challenge here as to how to classify texts without any pre-defined aspects.The existence of this kind of "misc" aspect text segments is very common in review corpora.It is difficult, even for domain experts, to nominate seed words for the "misc" aspect, which makes existing seed-driven text classification methods not applicable.Therefore, we propose to jointly model pre-defined aspects and the "misc" aspect through a novel framework, ARYA.It enables mutual enhancements between pre-defined aspects and the "misc" aspect via iterative classifier training and seed set updating.Specifically, it trains a classifier for pre-defined aspects and then leverages it to induce the supervision for the "misc" aspect.The prediction results of the "misc" aspect are later utilized to further filter the seed word selections for pre-defined aspects.Experiments in three domains demonstrate the superior performance of our proposed framework, as well as the necessity and importance of properly modeling the "misc" aspect. Jingbo Shang |
SDM | 3 |
| 2021 | UniTS: Short-Time Fourier Inspired Neural Networks for Sensory Time Series ClassificationabstractDiscovering patterns in time series data is essential to many key tasks in intelligent sensing systems, such as human activity recognition and event detection. These tasks involve the classification of sensory information from physical measurements such as inertial or temperature change measurements. Due to differences in the underlying physics, existing methods for classification use handcrafted features combined with traditional learning algorithms, or employ distinct deep neural models to directly learn from raw data. Shuheng Li, Ranak Roy Chowdhury, Jingbo Shang, Rajesh K. Gupta 0001, Dezhi Hong |
SenSys | 3 |
| 2021 | AutoName: A Corpus-Based Set Naming FrameworkabstractWe propose AutoName, an unsupervised framework that extracts a name for a set of query entities from a large-scale text corpus. Entity-set naming is useful in many tasks related to natural language processing and information retrieval such as session-based and conversational information seeking. Previous studies mainly extract set names from knowledge bases which provide highly reliable entity relations, but suffer from limited coverage of entities and set names that represent broad semantic classes. To address these problems, AutoName generates hypernym-anchored candidate phrases via probing a pre-trained language model and the entities' context in documents. Phrases are then clustered to identify ones that describe common concepts among query entities. Finally, AutoName ranks refined phrases based on the co-occurrences of their words with query entities and the conceptual integrity of their respective clusters. We built a new benchmark dataset for this task, consisting of 130 entity sets with name labels. Experimental results show that AutoName generates coherent and meaningful set names and significantly outperforms all baselines. Zhiqi Huang 0002, Razieh Rahimi, Puxuan Yu, Jingbo Shang, James Allan 0001 |
SIGIR | 4 |
| 2021 | Minimally-Supervised Structure-Rich Text Categorization via Learning on Text-Rich NetworksabstractText categorization is an essential task in Web content analysis. Considering the ever-evolving Web data and new emerging categories, instead of the laborious supervised setting, in this paper, we focus on the minimally-supervised setting that aims to categorize documents effectively, with a couple of seed documents annotated per category. We recognize that texts collected from the Web are often structure-rich, i.e., accompanied by various metadata. One can easily organize the corpus into a text-rich network, joining raw text documents with document attributes, high-quality phrases, label surface names as nodes, and their associations as edges. Such a network provides a holistic view of the corpus’ heterogeneous data sources and enables a joint optimization for network-based analysis and deep textual model training. We therefore propose a novel framework for minimally supervised categorization by learning from the text-rich network. Specifically, we jointly train two modules with different inductive biases – a text analysis module for text understanding and a network learning module for class-discriminative, scalable network learning. Each module generates pseudo training labels from the unlabeled document set, and both modules mutually enhance each other by co-training using pooled pseudo labels. We test our model on two real-world datasets. On the challenging e-commerce product categorization dataset with 683 categories, our experiments show that given only three seed documents per category, our framework can achieve an accuracy of about 92%, significantly outperforming all compared methods; our accuracy is only less than 2% away from the supervised BERT model trained on about 50K labeled documents. Xinyang Zhang 0002, Xin Dong 0001, Jingbo Shang, Jiawei Han 0001 |
WWW | 4 |
| 2020 | Contextualized Weak Supervision for Text ClassificationabstractWeakly supervised text classification based on a few user-provided seed words has recently attracted much attention from researchers.Existing methods mainly generate pseudo-labels in a context-free manner (e.g., string matching), therefore, the ambiguous, context-dependent nature of human language has been long overlooked.In this paper, we propose a novel framework ConWea, providing contextualized weak supervision for text classification.Specifically, we leverage contextualized representations of word occurrences and seed word information to automatically differentiate multiple interpretations of the same word, and thus create a contextualized corpus.This contextualized corpus is further utilized to train the classifier and expand seed words in an iterative manner.This process not only adds new contextualized, highly label-indicative keywords but also disambiguates initial seed words, making our weak supervision fully contextualized.Extensive experiments and case studies on real-world datasets demonstrate the necessity and significant advantages of using contextualized weak supervision, especially when the class labels are fine-grained. Dheeraj Mekala, Jingbo Shang |
ACL | 2 |
| 2020 | Empower Entity Set Expansion via Language Model ProbingabstractEntity set expansion, aiming at expanding a small seed entity set with new entities belonging to the same semantic class, is a critical task that benefits many downstream NLP and IR applications, such as question answering, query understanding, and taxonomy construction.Existing set expansion methods bootstrap the seed entity set by adaptively selecting context features and extracting new entities.A key challenge for entity set expansion is to avoid selecting ambiguous context features which will shift the class semantics and lead to accumulative errors in later iterations.In this study, we propose a novel iterative set expansion framework that leverages automatically generated class names to address the semantic drift issue.In each iteration, we select one positive and several negative class names by probing a pre-trained language model, and further score each candidate entity based on selected class names.Experiments on two datasets show that our framework generates high-quality class names and outperforms previous state-of-the-art methods significantly. Yunyi Zhang 0001, Jingbo Shang, Jiawei Han 0001 |
ACL | 3 |
| 2020 | META: Metadata-Empowered Weak Supervision for Text ClassificationabstractRecent advances in weakly supervised learning enable training high-quality text classifiers by only providing a few user-provided seed words.Existing methods mainly use text data alone to generate pseudo-labels despite the fact that metadata information (e.g., author and timestamp) is widely available across various domains.Strong label indicators exist in the metadata and it has been long overlooked mainly due to the following challenges: (1) metadata is multi-typed, requiring systematic modeling of different types and their combinations, (2) metadata is noisy, some metadata entities (e.g., authors, venues) are more compelling label indicators than others.In this paper, we propose a novel framework, META, which goes beyond the existing paradigm and leverages metadata as an additional source of weak supervision.Specifically, we organize the text data and metadata together into a text-rich network and adopt network motifs to capture appropriate combinations of metadata.Based on seed words, we rank and filter motif instances to distill highly label-indicative ones as "seed motifs", which provide additional weak supervision.Following a bootstrapping manner, we train the classifier and expand the seed words and seed motifs iteratively.Extensive experiments and case studies on real-world datasets demonstrate superior performance and significant advantages of leveraging metadata as weak supervision. Dheeraj Mekala, Xinyang Zhang 0002, Jingbo Shang |
EMNLP (1) | 3 |
| 2020 | SynSetExpan: An Iterative Framework for Joint Entity Set Expansion and Synonym DiscoveryabstractEntity set expansion and synonym discovery are two critical NLP tasks.Previous studies accomplish them separately, without exploring their interdependences.In this work, we hypothesize that these two tasks are tightly coupled because two synonymous entities tend to have similar likelihoods of belonging to various semantic classes.This motivates us to design SynSetExpan, a novel framework that enables two tasks to mutually enhance each other.SynSetExpan uses a synonym discovery model to include popular entities' infrequent synonyms into the set, which boosts the set expansion recall.Meanwhile, the set expansion model, being able to determine whether an entity belongs to a semantic class, can generate pseudo training data to fine-tune the synonym discovery model towards better accuracy.To facilitate the research on studying the interplays of these two tasks, we create the first large-scale Synonym-Enhanced Set Expansion (SE2) dataset via crowdsourcing.Extensive experiments on the SE2 dataset and previous benchmarks demonstrate the effectiveness of SynSetExpan for both entity set expansion and synonym discovery tasks. Wenda Qiu, Jingbo Shang, Michelle Vanni, Xiang Ren 0001, Jiawei Han 0001 |
EMNLP (1) | 3 |
| 2020 | Towards Adaptive Residual Network Training: A Neural-ODE PerspectiveabstractIn pursuit of resource-economical machine learning, attempts have been made to dynamically adjust computation workloads in different training stages, i.e., starting with a shallow network and gradually increasing the model depth (and computation workloads) during training. However, there is neither guarantee nor guidance on designing such network grow, due to the lack of its theoretical underpinnings. In this work, to explore the theory behind, we conduct theoretical analyses from an ordinary differential equation perspective. Specifically, we illustrate the dynamics of network growth and propose a novel performance measure specific to the depth increase. Illuminated by our analyses, we move towards theoretically sound growing operations and schedulers, giving rise to an adaptive training algorithm for residual networks, LipGrow, which automatically increases network depth thus accelerates training. In our experiments, it achieves comparable performance while reducing ∼ 50% of training time. Chengyu Dong, Zichao Li 0002, Jingbo Shang |
ICML | 4 |
| 2020 | Scientific Text Mining and Knowledge GraphsabstractUnstructured scientific text, in various forms of textual artifacts, including manuscripts, publications, patents, and proposals, is used to store the tremendous wealth of knowledge discovered after weeks, months, and years, developing hypotheses, working in the lab or clinic, and analyzing results. A grand challenge on data mining research is to develop effective methods for transforming the scientific text into well-structured forms (e.g., ontology, taxonomy, knowledge graphs), so that machine intelligent systems can build on them for hypothesis generation and validation. In this tutorial, we provide a comprehensive overview on recent research and development in this direction. First, we introduce a series of text mining methods that extract phrases, entities, scientific concepts, relations, claims, and experimental evidence. Then we discuss methods that construct and learn from scientific knowledge graphs for accurate search, document classification, and exploratory analysis. Specifically, we focus on scalable, effective, weakly supervised methods that work on text in sciences (e.g., chemistry, biology). Meng Jiang 0001, Jingbo Shang |
KDD | 2 |
| 2020 | FUSE: Multi-faceted Set Expansion by Coherent Clustering of Skip-Grams
Wanzheng Zhu, Hongyu Gong, Chao Zhang 0014, Jingbo Shang, Suma Bhat, Jiawei Han 0001 |
ECML/PKDD (3) | 5 |
| 2020 | NetTaxo: Automated Topic Taxonomy Construction from Text-Rich NetworkabstractThe automated construction of topic taxonomies can benefit numerous applications, including web search, recommendation, and knowledge discovery. One of the major advantages of automatic taxonomy construction is the ability to capture corpus-specific information and adapt to different scenarios. To better reflect the characteristics of a corpus, we take the meta-data of documents into consideration and view the corpus as a text-rich network. In this paper, we propose NetTaxo, a novel automatic topic taxonomy construction framework, which goes beyond the existing paradigm and allows text data to collaborate with network structure. Specifically, we learn term embeddings from both text and network as contexts. Network motifs are adopted to capture appropriate network contexts. We conduct an instance-level selection for motifs, which further refines term embedding according to the granularity and semantics of each taxonomy node. Clustering is then applied to obtain sub-topics under a taxonomy node. Extensive experiments on two real-world datasets demonstrate the superiority of our method over the state-of-the-art, and further verify the effectiveness and importance of instance-level motif selection. Jingbo Shang, Xinyang Zhang 0002, Jiawei Han 0001 |
WWW | 1 |
| 2019 | Distantly Supervised Biomedical Named Entity Recognition with Dictionary ExpansionabstractState-of-the-art biomedical named entity recognition (BioNER) systems apply supervised machine learning models (i.e., relying on human effort for training data annotation) which are not easy to be generalized to new entity types and datasets. We propose a distantly supervised approach, AutoBioNER, that automatically recognizes biomedical entities from massive corpora with user-input dictionaries. AutoBioNER does not need any human annotated data. It relies on incomplete entity dictionaries to provide seeds for each entity type and performs a novel entity set expansion step for corpus-level new entity recognition and dictionary completion. The expanded dictionaries are used as distant supervision to train a neural model for BioNER. Experimental results show that AutoBioNER achieves the best performance among the methods that only use dictionaries with no additional human effort on BioNER benchmark datasets. It is also demonstrated that the dictionary expansion step plays an important role in the great performances. Xuan Wang 0008, Yu Zhang 0044, Qi Li 0012, Xiang Ren 0001, Jingbo Shang, Jiawei Han 0001 |
BIBM | 5 |
| 2019 | CrossWeigh: Training Named Entity Tagger from Imperfect AnnotationsabstractZihan Wang, Jingbo Shang, Liyuan Liu, Lihao Lu, Jiacheng Liu, Jiawei Han. Proceedings of the 2019 Conference on Empirical Methods in Natural Language Processing and the 9th International Joint Conference on Natural Language Processing (EMNLP-IJCNLP). 2019. Zihan Wang 0001, Jingbo Shang, Lihao Lu, Jiacheng Liu 0004, Jiawei Han 0001 |
EMNLP/IJCNLP (1) | 2 |
| 2019 | Constructing and Mining Heterogeneous Information Networks from Massive TextabstractReal-world data exists largely in the form of unstructured texts. A grand challenge on data mining research is to develop effective and scalable methods that may transform unstructured text into structured knowledge. Based on our vision, it is highly beneficial to transform such text into structured heterogeneous information networks, on which actionable knowledge can be generated based on the user's need. In this tutorial, we provide a comprehensive overview on recent research and development in this direction. First, we introduce a series of effective methods that construct heterogeneous information networks from massive, domain-specific text corpora. Then we discuss methods that mine such text-rich networks based on the user's need. Specifically, we focus on scalable, effective, weakly supervised, language-agnostic methods that work on various kinds of text. We further demonstrate, on real datasets (including news articles, scientific publications, and product reviews), how information networks can be constructed and how they can assist further exploratory analysis. Jingbo Shang, Jiawei Han 0001 |
KDD | 1 |
| 2019 | Integrating Local Context and Global Cohesiveness for Open Information ExtractionabstractExtracting entities and their relations from text is an important task for understanding massive text corpora. Open information extraction (IE) systems mine relation tuples (i.e., entity arguments and a predicate string to describe their relation) from sentences. These relation tuples are not confined to a predefined schema for the relations of interests. However, current Open IE systems focus on modeling local context information in a sentence to extract relation tuples, while ignoring the fact that global statistics in a large corpus can be collectively leveraged to identify high-quality sentence-level extractions. In this paper, we propose a novel Open IE system, called ReMine, which integrates local context signals and global structural signals in a unified, distant-supervision framework. Leveraging facts from external knowledge bases as supervision, the new system can be applied to many different domains to facilitate sentence-level tuple extractions using corpus-level statistics. Our system operates by solving a joint optimization problem to unify (1) segmenting entity/relation phrases in individual sentences based on local context; and (2) measuring the quality of tuples extracted from individual sentences with a translating-based objective. Learning the two subtasks jointly helps correct errors produced in each subtask so that they can mutually enhance each other. Experiments on two real-world corpora from different domains demonstrate the effectiveness, generality, and robustness of ReMine when compared to state-of-the-art open IE systems. Qi Zhu 0008, Xiang Ren 0001, Jingbo Shang, Yu Zhang 0044, Ahmed El-Kishky, Jiawei Han 0001 |
WSDM | 3 |
| 2019 | Cross-type biomedical named entity recognition with deep multi-task learningabstractMOTIVATION: State-of-the-art biomedical named entity recognition (BioNER) systems often require handcrafted features specific to each entity type, such as genes, chemicals and diseases. Although recent studies explored using neural network models for BioNER to free experts from manual feature engineering, the performance remains limited by the available training data for each entity type. RESULTS: We propose a multi-task learning framework for BioNER to collectively use the training data of different types of entities and improve the performance on each of them. In experiments on 15 benchmark BioNER datasets, our multi-task model achieves substantially better performance compared with state-of-the-art BioNER systems and baseline neural sequence labeling models. Further analysis shows that the large performance gains come from sharing character- and word-level information among relevant biomedical entities across differently labeled corpora. AVAILABILITY AND IMPLEMENTATION: Our source code is available at https://github.com/yuzhimanhua/lm-lstm-crf. SUPPLEMENTARY INFORMATION: Supplementary data are available at Bioinformatics online. Xuan Wang 0008, Yu Zhang 0044, Xiang Ren 0001, Yuhao Zhang 0004, Marinka Zitnik, Jingbo Shang, Curt Langlotz, Jiawei Han 0001 |
Bioinform. | 6 |
| 2019 | TextCube: Automated Construction and Multidimensional ExplorationabstractToday's society is immersed in a wealth of text data, ranging from news articles, to social media, research literature, medical records, and corporate reports. A grand challenge of data science and engineering is to develop effective and scalable methods to extract structures and knowledge from massive text data to satisfy diverse applications, without extensive, corpus-specific human annotations. In this tutorial, we show that TextCube provides a critical information organization structure that will satisfy such an information need. We overview a set of recently developed data-driven methods that facilitate automated construction of TextCubes from massive, domain-specific text corpora, and show that TextCubes so constructed will enhance text exploration and analysis for various applications. We focus on new TextCube construction methods that are scalable, weakly-supervised, domain-independent, language-agnostic, and effective (i.e., generating quality TextCubes from large corpora of various domains). We will demonstrate with real datasets (including news articles, scientific publications, and product reviews) on how TextCubes can be constructed to assist multidimensional analysis of massive text corpora. Yu Meng 0001, Jiaxin Huang 0001, Jingbo Shang, Jiawei Han 0001 |
Proc. VLDB Endow. | 3 |
| 2018 | Empower Sequence Labeling with Task-Aware Neural Language ModelabstractLinguistic sequence labeling is a general approach encompassing a variety of problems, such as part-of-speech tagging and named entity recognition. Recent advances in neural networks (NNs) make it possible to build reliable models without handcrafted features. However, in many cases, it is hard to obtain sufficient annotations to train these models. In this study, we develop a neural framework to extract knowledge from raw texts and empower the sequence labeling task. Besides word-level knowledge contained in pre-trained word embeddings, character-aware neural language models are incorporated to extract character-level knowledge. Transfer learning techniques are further adopted to mediate different components and guide the language model towards the key knowledge. Comparing to previous methods, these task-specific knowledge allows us to adopt a more concise model and conduct more efficient training. Different from most transfer learning methods, the proposed framework does not rely on any additional supervision. It extracts knowledge from self-contained order information of training sequences. Extensive experiments on benchmark datasets demonstrate the effectiveness of leveraging character-level knowledge and the efficiency of co-training. For example, on the CoNLL03 NER task, model training completes in about 6 hours on a single GPU, reaching F_1 score of 91.71+/-0.10 without using any extra annotations. Jingbo Shang, Xiang Ren 0001, Frank F. Xu, Huan Gui, Jian Peng 0001, Jiawei Han 0001 |
AAAI | 2 |
| 2018 | Investigating Rumor News Using Agreement-Aware SearchabstractRecent years have witnessed a widespread increase of rumor news generated by humans and machines. Therefore, tools for investigating rumor news have become an urgent necessity. One useful function of such tools is to see ways a specific topic or event is represented by presenting different points of view from multiple sources. In this paper, we propose Maester, a novel agreement-aware search framework for investigating rumor news. Given an investigative question, Maester will retrieve related articles to that question, assign and display top articles from agree, disagree, and discuss categories to users. Splitting the results into these three categories provides the user a holistic view towards the investigative question. We build Maester based on the following two key observations: (1) relatedness can commonly be determined by keywords and entities occurring in both questions and articles, and (2) the level of agreement between the investigative question and the related news article can often be decided by a few key sentences. Accordingly, we use gradient boosting tree models with keyword/entity matching features for relatedness detection, and leverage recurrent neural network to infer the level of agreement. Our experiments on the Fake News Challenge (FNC) dataset demonstrate up to an order of magnitude improvement of Maester over the original FNC winning solution, for agreement-aware search. Jingbo Shang, Tianhang Sun, Xingbang Liu, Anja Gruenheid, Flip Korn, Ádám Dániel Lelkes, Cong Yu 0001, Jiawei Han 0001 |
CIKM | 1 |
| 2018 | Efficient Contextualized Representation: Language Model Pruning for Sequence LabelingabstractMany efforts have been made to facilitate natural language processing tasks with pre-trained language models (LMs), and brought significant improvements to various applications.To fully leverage the nearly unlimited corpora and capture linguistic information of multifarious levels, large-size LMs are required; but for a specific task, only parts of these information are useful.Such large-sized LMs, even in the inference stage, may cause heavy computation workloads, making them too time-consuming for large-scale applications.Here we propose to compress bulky LMs while preserving useful information with regard to a specific task.As different layers of the model keep different information, we develop a layer selection method for model pruning using sparsityinducing regularization.By introducing the dense connectivity, we can detach any layer without affecting others, and stretch shallow and wide LMs to be deep and narrow.In model training, LMs are learned with layerwise dropouts for better robustness.Experiments on two benchmark datasets demonstrate the effectiveness of our method. Xiang Ren 0001, Jingbo Shang, Xiaotao Gu, Jian Peng 0001, Jiawei Han 0001 |
EMNLP | 3 |
| 2018 | Learning Named Entity Tagger using Domain-Specific DictionaryabstractRecent advances in deep neural models allow us to build reliable named entity recognition (NER) systems without handcrafting features.However, such methods require large amounts of manually-labeled training data.There have been efforts on replacing human annotations with distant supervision (in conjunction with external dictionaries), but the generated noisy labels pose significant challenges on learning effective neural models.Here we propose two neural models to suit noisy distant supervision from the dictionary.First, under the traditional sequence labeling framework, we propose a revised fuzzy CRF layer to handle tokens with multiple possible labels.After identifying the nature of noisy labels in distant supervision, we go beyond the traditional framework and propose a novel, more effective neural model AutoNER with a new Tie or Break scheme.In addition, we discuss how to refine distant supervision for better NER performance.Extensive experiments on three benchmark datasets demonstrate that AutoNER achieves the best performance when only using dictionaries with no additional human effort, and delivers competitive results with state-of-the-art supervised benchmarks. Jingbo Shang, Xiaotao Gu, Xiang Ren 0001, Teng Ren, Jiawei Han 0001 |
EMNLP | 1 |
| 2018 | Entity Set Search of Scientific Literature: An Unsupervised Ranking ApproachabstractLiterature search is critical for any scientific research. Different from Web or general domain search, a large portion of queries in scientific literature search are entity-set queries, that is, multiple entities of possibly different types. Entity-set queries reflect user's need for finding documents that contain multiple entities and reveal inter-entity relationships and thus pose non-trivial challenges to existing search algorithms that model each entity separately. However, entity-set queries are usually sparse (i.e., not so repetitive), which makes ineffective many supervised ranking models that rely heavily on associated click history. To address these challenges, we introduce SetRank, an unsupervised ranking framework that models inter-entity relationships and captures entity type information. Furthermore, we develop a novel unsupervised model selection algorithm, based on the technique of weighted rank aggregation, to automatically choose the parameter settings in SetRank without resorting to a labeled validation set. We evaluate our proposed unsupervised approach using datasets from TREC Genomics Tracks and Semantic Scholar's query log. The experiments demonstrate that SetRank significantly outperforms the baseline unsupervised models, especially on entity-set queries, and our model selection algorithm effectively chooses suitable parameter settings. Jinfeng Xiao, Jingbo Shang, Saurabh Sinha 0002, Jiawei Han 0001 |
SIGIR | 4 |
| 2018 | DPPred: An Effective Prediction Framework with Concise Discriminative PatternsabstractIn the literature, two series of models have been proposed to address prediction problems including classification and regression. Simple models, such as generalized linear models, have ordinary performance but strong interpretability on a set of simple features. The other series, including tree-based models, organize numerical, categorical and high dimensional features into a comprehensive structure with rich interpretable information in the data. In this paper, we propose a novel Discriminative Pattern-based Prediction framework (DPPred) to accomplish the prediction tasks by taking their advantages of both effectiveness and interpretability. Specifically, DPPred adopts the concise discriminative patterns that are on the prefix paths from the root to leaf nodes in the tree-based models. DPPred selects a limited number of the useful discriminative patterns by searching for the most effective pattern combination to fit generalized linear models. Extensive experiments show that in many scenarios, DPPred provides competitive accuracy with the state-of-the-art as well as the valuable interpretability for developers and experts. In particular, taking a clinical application dataset as a case study, our DPPred outperforms the baselines by using only 40 concise discriminative patterns out of a potentially exponentially large set of patterns. Jingbo Shang, Meng Jiang 0001, Wenzhu Tong, Jinfeng Xiao, Jian Peng 0001, Jiawei Han 0001 |
IEEE Trans. Knowl. Data Eng. | 1 |
| 2018 | Automated Phrase Mining from Massive Text CorporaabstractAs one of the fundamental tasks in text analysis, phrase mining aims at extracting quality phrases from a text corpus and has various downstream applications including information extraction/retrieval, taxonomy construction, and topic modeling. Most existing methods rely on complex, trained linguistic analyzers, and thus likely have unsatisfactory performance on text corpora of new domains and genres without extra but expensive adaption. None of the state-of-the-art models, even data-driven models, is fully automated because they require human experts for designing rules or labeling phrases. In this paper, we propose a novel framework for automated phrase mining, AutoPhrase, which supports any language as long as a general knowledge base (e.g., Wikipedia) in that language is available, while benefiting from, but not requiring, a POS tagger. Compared to the state-of-the-art methods, AutoPhrase has shown significant improvements in both effectiveness and efficiency on five real-world datasets across different domains and languages. Besides, AutoPhrase can be extended to model single-word quality phrases. Jingbo Shang, Meng Jiang 0001, Xiang Ren 0001, Clare R. Voss, Jiawei Han 0001 |
IEEE Trans. Knowl. Data Eng. | 1 |
| 2017 | An Attention-based Collaboration Framework for Multi-View Network Representation LearningabstractLearning distributed node representations in networks has been attracting increasing attention recently due to its effectiveness in a variety of applications. Existing approaches usually study networks with a single type of proximity between nodes, which defines a single view of a network. However, in reality there usually exists multiple types of proximities between nodes, yielding networks with multiple views. This paper studies learning node representations for networks with multiple views, which aims to infer robust node representations across different views. We propose a multi-view representation learning approach, which promotes the collaboration of different views and lets them vote for the robust representations. During the voting process, an attention mechanism is introduced, which enables each node to focus on the most informative views. Experimental results on real-world networks show that the proposed approach outperforms existing state-of-the-art approaches for network representation learning with a single view and other competitive approaches with multiple views. Meng Qu, Jian Tang 0005, Jingbo Shang, Xiang Ren 0001, Ming Zhang 0004, Jiawei Han 0001 |
CIKM | 3 |
| 2017 | Detecting Multiple Periods and Periodic Patterns in Event Time SequencesabstractPeriodicity is prevalent in physical world, and many events involve more than one periods, eg individual's mobility, tide pattern, and massive transportation utilization. Knowing the true periods of events can benefit a number of applications, such as traffic prediction, time-aware recommendation and advertisement, and anomaly detection. However, detecting multiple periods is a very challenging task due to not only the interwoven periodic patterns but also the low quality of event tracking records. In this paper, we study the problem of discovering all true periods and the corresponded occurring patterns of an event from a noisy and incomplete observation sequence. We devise a novel scoring function, by maximizing which we can identify the true periodic patterns involved in the sequence. We prove that, however, optimizing the objective function is an NP-hard problem. To address this challenge, we develop a heuristic algorithm named Timeslot Coverage Model (TiCom), for identifying the periods and periodic patterns approximately. The results of extensive experiments on both synthetic and real-life datasets show that our model outperforms the state-of-the-art baselines significantly in various tasks, including period detection, periodic pattern identification, and anomaly detection. Quan Yuan 0001, Jingbo Shang, Xin Cao 0001, Chao Zhang 0014, Xinhe Geng, Jiawei Han 0001 |
CIKM | 2 |
| 2017 | MetaPAD: Meta Pattern Discovery from Massive Text CorporaabstractMining textual patterns in news, tweets, papers, and many other kinds of text corpora has been an active theme in text mining and NLP research. Previous studies adopt a dependency parsing-based pattern discovery approach. However, the parsing results lose rich context around entities in the patterns, and the process is costly for a corpus of large scale. In this study, we propose a novel typed textual pattern structure, called meta pattern, which is extended to a frequent, informative, and precise subsequence pattern in certain context. We propose an efficient framework, called MetaPAD, which discovers meta patterns from massive corpora with three techniques: (1) it develops a context-aware segmentation method to carefully determine the boundaries of patterns with a learnt pattern quality assessment function, which avoids costly dependency parsing and generates high-quality patterns; (2) it identifies and groups synonymous meta patterns from multiple facets---their types, contexts, and extractions; and (3) it examines type distributions of entities in the instances extracted by each group of patterns, and looks for appropriate type levels to make discovered patterns precise. Experiments demonstrate that our proposed framework discovers high-quality typed textual patterns efficiently from different genres of massive corpora and facilitates information extraction. Meng Jiang 0001, Jingbo Shang, Taylor Cassidy, Xiang Ren 0001, Lance M. Kaplan, Tim Hanratty, Jiawei Han 0001 |
KDD | 2 |
| 2017 | SetExpan: Corpus-Based Set Expansion via Context Feature Selection and Rank Ensemble
Zeqiu Wu, Dongming Lei, Jingbo Shang, Xiang Ren 0001, Jiawei Han 0001 |
ECML/PKDD (1) | 4 |
| 2017 | Building Structured Databases of Factual Knowledge from Massive Text CorporaabstractIn today's computerized and information-based society, people are inundated with vast amounts of text data, ranging from news articles, social media post, scientific publications, to a wide range of textual information from various domains (corporate reports, advertisements, legal acts, medical reports). To turn such massive unstructured text data into structured, actionable knowledge, one of the grand challenges is to gain an understanding of the factual information (e.g., entities, attributes, relations) in the text. Xiang Ren 0001, Meng Jiang 0001, Jingbo Shang, Jiawei Han 0001 |
SIGMOD Conference | 3 |
| 2016 | Data-Driven Contextual Valence Shifter Quantification for Multi-Theme Sentiment AnalysisabstractUsers often write reviews on different themes involving linguistic structures with complex sentiments. The sentiment polarity of a word can be different across themes. Moreover, contextual valence shifters may change sentiment polarity depending on the contexts that they appear in. Both challenges cannot be modeled effectively and explicitly in traditional sentiment analysis. Studying both phenomena requires multi-theme sentiment analysis at the word level, which is very interesting but significantly more challenging than overall polarity classification. To simultaneously resolve the multi-theme and sentiment shifting problems, we propose a data-driven framework to enable both capabilities: (1) polarity predictions of the same word in reviews of different themes, and (2) discovery and quantification of contextual valence shifters. The framework formulates multi-theme sentiment by factorizing the review sentiments with theme/word embeddings and then derives the shifter effect learning problem as a logistic regression. The improvement of sentiment polarity classification accuracy demonstrates not only the importance of multi-theme and sentiment shifting, but also effectiveness of our framework. Human evaluations and case studies further show the success of multi-theme word sentiment predictions and automatic effect quantification of contextual valence shifters. Hongkun Yu 0001, Jingbo Shang, Meichun Hsu, Malú Castellanos, Jiawei Han 0001 |
CIKM | 2 |
| 2016 | MACFP: Maximal Approximate Consecutive Frequent Pattern Mining under Edit DistanceabstractConsecutive pattern mining aiming at finding sequential patterns substrings, is a special case of frequent pattern mining and has been played a crucial role in many real world applications, especially in biological sequence analysis, time series analysis, and network log mining. Approximations, including insertions, deletions, and substitutions, between strings are widely used in biological sequence comparisons. However, most existing string pattern mining methods only consider hamming distance without insertions/deletions (indels). Little attention has been paid to the general approximate consecutive frequent pattern mining under edit distance, potentially due to the high computational complexity, particularly on DNA sequences with billions of base pairs. In this paper, we introduce an efficient solution to this problem. We first formulate the Maximal Approximate Consecutive Frequent Pattern Mining (MACFP) problem that identifies substring patterns under edit distance in a long query sequence. Then, we propose a novel algorithm with linear time complexity to check whether the support of a substring pattern is above a predefined threshold in the query sequence, thus greatly reducing the computational complexity of MACFP. With this fast decision algorithm, we can efficiently solve the original pattern discovery problem with several indexing and searching techniques. Comprehensive experiments on sequence pattern analysis and a study on cancer genomics application demonstrate the effectiveness and efficiency of our algorithm, compared to several existing methods. Jingbo Shang, Jian Peng 0001, Jiawei Han 0001 |
SDM | 1 |
| 2016 | DPClass: An Effective but Concise Discriminative Patterns-Based Classification FrameworkabstractPattern-based classification was originally proposed to improve the accuracy using selected frequent patterns, where many efforts were paid to prune a huge number of non-discriminative frequent patterns. On the other hand, tree-based models have shown strong abilities on many classification tasks since they can easily build high-order interactions between different features and also handle both numerical and categorical features as well as high dimensional features. By taking the advantage of both modeling methodologies, we propose a natural and effective way to resolve pattern-based classification by adopting discriminative patterns which are the prefix paths from root to nodes in tree-based models (e.g., random forest). Moreover, we further compress the number of discriminative patterns by selecting the most effective pattern combinations that fit into a generalized linear model. As a result, our discriminative pattern-based classification framework (DPClass) could perform as good as previous state-of-the-art algorithms, provide great interpretability by utilizing only very limited number of discriminative patterns, and predict new data extremely fast. More specifically, in our experiments, DPClass could gain even better accuracy by only using top-20 discriminative patterns. The framework so generated is very concise and highly explanatory to human experts. Jingbo Shang, Wenzhu Tong, Jian Peng 0001, Jiawei Han 0001 |
SDM | 1 |
| 2016 | Representing Documents via Latent Keyphrase InferenceabstractMany text mining approaches adopt bag-of-words or $n$-grams models to represent documents. Looking beyond just the words, fiie, the explicit surface forms, in a document can improve a computer's understanding of text. Being aware of this, researchers have proposed concept-based models that rely on a human-curated knowledge base to incorporate other related concepts in the document representation. But these methods are not desirable when applied to vertical domains (eg, literature, enterprise, etc) due to low coverage of in-domain concepts in the general knowledge base and interference from out-of-domain concepts. In this paper, we propose a data-driven model named Latent Keyphrase Inference LAKI) that represents documents with a vector of closely related domain keyphrases instead of single words or existing concepts in the knowledge base. We show that given a corpus of in-domain documents, topical content units can be learned for each domain keyphrase, which enables a computer to do smart inference to discover latent document keyphrases, going beyond just explicit mentions. Compared with the state-of-art document representation approaches, LAKI fills the gap between bag-of-words and concept-based models by using domain keyphrases as the basic representation unit. It removes dependency on a knowledge base while providing, with keyphrases, readily interpretable representations. When evaluated against 8 other methods on two text mining tasks over two corpora, LAKI outperformed all. Xiang Ren 0001, Jingbo Shang, Taylor Cassidy, Clare R. Voss, Jiawei Han 0001 |
WWW | 3 |
| 2015 | Detecting urban black holes based on human mobility dataabstractMany types of human mobility data, such as flows of taxicabs, card swiping data of subways, bike trip data and Call Details Records (CDR), can be modeled by a Spatio-Temporal Graph (STG). STG is a directed graph in which vertices and edges are associated with spatio-temporal properties (e.g. the traffic flow on a road and the geospatial location of an intersection). In this paper, we instantly detect interesting phenomena, entitled black holes and volcanos, from an STG. Specifically, a black hole is a subgraph (of an STG) that has the overall inflow greater than the overall outflow by a threshold, while a volcano is a subgraph with the overall outflow greater than the overall inflow by a threshold (detecting volcanos from an STG is proved to be equivalent to the detection of black holes). The online detection of black holes/volcanos can timely reflect anomalous events, such as disasters, catastrophic accidents, and therefore help keep public safety. The patterns of black holes/volcanos and the relations between them reveal human mobility patterns in a city, thus help formulate a better city planning or improve a system's operation efficiency. Based on a well-designed STG index, we propose a two-step black hole detection algorithm: The first step identifies a set of candidate grid cells to start from; the second step expands an initial edge in a candidate cell to a black hole and prunes other candidate cells after a black hole is detected. Then, we adapt this detection algorithm to a continuous black hole detection scenario. We evaluate our method based on Beijing taxicab data and the bike trip data in New York, finding urban anomalies and human mobility patterns. Liang Hong 0001, Yu Zheng 0004, Duncan Yung, Jingbo Shang, Lei Zou 0001 |
SIGSPATIAL/GIS | 4 |
| 2015 | Mining Quality Phrases from Massive Text CorporaabstractText data are ubiquitous and play an essential role in big data applications. However, text data are mostly unstructured. Transforming unstructured text into structured units (e.g., semantically meaningful phrases) will substantially reduce semantic ambiguity and enhance the power and efficiency at manipulating such data using database technology. Thus mining quality phrases is a critical research problem in the field of databases. In this paper, we propose a new framework that extracts quality phrases from text corpora integrated with phrasal segmentation. The framework requires only limited training but the quality of phrases so generated is close to human judgment. Moreover, the method is scalable: both computation time and required space grow linearly as corpus size increases. Our experiments on large text corpora demonstrate the quality and efficiency of the new method. Jingbo Shang, Chi Wang 0001, Xiang Ren 0001, Jiawei Han 0001 |
SIGMOD Conference | 2 |
| 2014 | Constrained Question Recommendation in MOOCs via SubmodularityabstractA recent area in which recommender systems have shown their value is in online discussion forums and question-answer sites. Earlier work in this space has focused on the problem of matching participants to opportunities but has not adequately addressed the problem that in these social contexts, multiple dimensions of constraints must be satisfied, including limitations on capacity and minimal requirements for expertise. In this work, we propose such a constrained question recommendation problem with load balance constraints in discussion forums and use flow based model to generate the optimal solution. In particular, to address the introduced computation complexity, we investigate the concept of submodularity of the objective function and propose a specific submodular method to give an approximated solution. We present experiments conducted on two Massive Open Online Course (MOOC) discussion forum datasets, and demonstrate the effectiveness and efficiency of our submodular method in solving constrained question recommendation tasks. Diyi Yang, Jingbo Shang, Carolyn P. Rosé |
CIKM | 2 |
| 2014 | A Parallel and Efficient Algorithm for Learning to MatchabstractMany tasks in data mining and related fields can be formalized as matching between objects in two heterogeneous domains, including collaborative filtering, link prediction, image tagging, and web search. Machine learning techniques, referred to as learning-to-match in this paper, have been successfully applied to the problems. Among them, a class of state-of-the-art methods, named feature-based matrix factorization, formalize the task as an extension to matrix factorization by incorporating auxiliary features into the model. Unfortunately, making those algorithms scale to real world problems is challenging, and simple parallelization strategies fail due to the complex cross talking patterns between sub-tasks. In this paper, we tackle this challenge with a novel parallel and efficient algorithm. Our algorithm, based on coordinate descent, can easily handle hundreds of millions of instances and features on a single machine. The key recipe of this algorithm is an iterative relaxation of the objective to facilitate parallel updates of parameters, with guaranteed convergence on minimizing the original objective function. Experimental results demonstrate that the proposed method is effective on a wide range of matching problems, with efficiency significantly improved upon the baselines while accuracy retained unchanged. Jingbo Shang, Tianqi Chen 0001, Hang Li 0001, Zhengdong Lu, Yong Yu 0001 |
ICDM | 1 |
| 2014 | Inferring gas consumption and pollution emission of vehicles throughout a cityabstractThis paper instantly infers the gas consumption and pollution emission of vehicles traveling on a city's road network in a current time slot, using GPS trajectories from a sample of vehicles (e.g., taxicabs). The knowledge can be used to suggest cost-efficient driving routes as well as identifying road segments where gas has been wasted significantly. The instant estimation of the emissions from vehicles can enable pollution alerts and help diagnose the root cause of air pollution in the long run. In our method, we first compute the travel speed of each road segment using the GPS trajectories received recently. As many road segments are not traversed by trajectories (i.e., data sparsity), we propose a Travel Speed Estimation (TSE) model based on a context-aware matrix factorization approach. TSE leverages features learned from other data sources, e.g., map data and historical trajectories, to deal with the data sparsity problem. We then propose a Traffic Volume Inference (TVI) model to infer the number of vehicles passing each road segment per minute. TVI is an unsupervised Bayesian Network that incorporates multiple factors, such as travel speed, weather conditions and geographical features of a road. Given the travel speed and traffic volume of a road segment, gas consumption and emissions can be calculated based on existing environmental theories. We evaluate our method based on extensive experiments using GPS trajectories generated by over 32,000 taxis in Beijing over a period of two months. The results demonstrate the advantages of our method over baselines, validating the contribution of its components and finding interesting discoveries for the benefit of society. Jingbo Shang, Yu Zheng 0004, Wenzhu Tong, Eric Chang, Yong Yu 0001 |
KDD | 1 |