EDBT 2026 Demo / reviewers in the wild / expert
Zhenwei Dai
dblp:236/6549
· DBLP profile ↗
12ranked-venue papers
4as first author
9since 2021 · last 2025
—ORCID · conflict
Domains — the database's venue-derived domains; a paper can count in several
Artificial intelligence and machine learning · 7 · 1 first-author · 6 since 2021Databases, data management, data science and information retrieval · 4 · 1 first-author · 4 since 2021Applied, interdisciplinary, general and emerging computing · 3 · 2 first-author · 1 since 2021
| Year | Publication | Venue | Position |
|---|---|---|---|
| 2025 | The 2nd Workshop on Large Language Models for E-CommerceabstractLarge Language Models (LLMs) are revolutionizing E-Commerce by enabling product recommendation, search, classification, question answering, and advertising applications. Their increasing adoption in real-world systems underscores their potential; however, challenges persist in ensuring accuracy, efficiency, fairness, and privacy. This workshop aims to bring together researchers and industry practitioners to explore both the limitations and opportunities of LLMs in e-commerce. The workshop seeks to foster collaboration, bridge the gap between academia and industry, and drive innovation in the application of LLMs to E-Commerce through discussions on model design, algorithmic advancements, and practical deployment. Haoyu Han 0001, Fali Wang, Chen Luo 0003, Hui Liu 0031, Zhenwei Dai, Qi He 0002, Dawei Yin 0001, Suhang Wang, Jiliang Tang, Jian Pei 0001, Xianfeng Tang |
KDD (2) | 7 |
| 2025 | SimRAG: Self-Improving Retrieval-Augmented Generation for Adapting Large Language Models to Specialized DomainsabstractRan Xu, Hui Liu, Sreyashi Nag, Zhenwei Dai, Yaochen Xie, Xianfeng Tang, Chen Luo, Yang Li, Joyce C. Ho, Carl Yang, Qi He. 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. Ran Xu 0002, Hui Liu 0033, Sreyashi Nag, Zhenwei Dai, Yaochen Xie, Xianfeng Tang, Chen Luo 0003, Yang Li 0055, Joyce C. Ho, Carl Yang 0001, Qi He 0002 |
NAACL (Long Papers) | 4 |
| 2025 | Keeping an Eye on LLM Unlearning: The Hidden Risk and RemedyabstractAlthough Large Language Models (LLMs) have demonstrated impressive capabilities across a wide range of tasks, growing concerns have emerged over the misuse of sensitive, copyrighted, or harmful data during training. To address these concerns, unlearning techniques have been developed to remove the influence of specific data without retraining from scratch. However, this paper reveals a critical vulnerability in fine-tuning-based unlearning: a malicious user can craft a manipulated forgetting request that stealthily degrades the model’s utility for benign users. We demonstrate this risk through a red-teaming Stealthy Attack (SA), which is inspired by two key limitations of existing unlearning—the inability to constrain the scope of unlearning effect and the failure to distinguish benign tokens from unlearning signals. Prior work has shown that unlearned models tend to memorize forgetting data as unlearning signals, and respond with hallucinations or feigned ignorance when unlearning signals appear in the input. By subtly increasing the presence of common benign tokens in the forgetting data, SA enhances the connection between benign tokens and unlearning signals. As a result, when normal users include such tokens in their prompts, the model exhibits unlearning behaviors, leading to unintended utility degradation. To address this vulnerability, we propose Scope-aware Unlearning (SU), a lightweight enhancement that introduces a scope term into the unlearning objective, encouraging the model to localize the forgetting effect. Our method requires no additional data processing, integrates seamlessly with existing fine-tuning frameworks, and significantly improves robustness against SA. Extensive experiments validate the effectiveness of both SA and SU. Jie Ren 0019, Zhenwei Dai, Xianfeng Tang, Yue Xing 0002, Shenglai Zeng, Jingying Zeng, Qiankun Peng, Samarth Varshney, Suhang Wang, Qi He 0002, Charu C. Aggarwal, Hui Liu 0003 |
NeurIPS | 2 |
| 2025 | AgentTTS: Large Language Model Agent for Test-time Compute-optimal Scaling Strategy in Complex TasksabstractTest-time scaling (TTS) enhances the performance of large language models (LLMs) by allocating additional compute resources during inference. However, existing research primarily investigates TTS in single-stage tasks; while many real-world problems are multi-stage complex tasks, composed of a sequence of heterogeneous subtasks with each subtask requires LLM of specific capability. Therefore, we study a novel problem: the test-time compute-optimal scaling in multi-stage complex tasks, aiming to select suitable models and allocate budgets per subtask to maximize overall performance. TTS in multi-stage tasks introduces two fundamental challenges: (i) The combinatorial search space of model and budget allocations, combined with the high cost of inference, makes brute-force search impractical. (ii) The optimal model and budget allocations across subtasks are interdependent, increasing the complexity of the compute-optimal search. To address this gap, we conduct extensive pilot experiments on four tasks across six datasets, deriving three empirical insights characterizing the behavior of LLMs in multi-stage complex tasks. Informed by these insights, we propose AgentTTS, an LLM-agent-based framework that autonomously searches for compute-optimal allocations through iterative feedback-driven interactions with the execution environment. Experimental results demonstrate that AgentTTS significantly outperforms traditional and other LLM-based baselines in search efficiency, and shows improved robustness to varying training set sizes and enhanced interpretability. Fali Wang, Hui Liu 0033, Zhenwei Dai, Jingying Zeng, Zhiwei Zhang 0028, Zongyu Wu 0001, Chen Luo 0003, Xianfeng Tang, Qi He 0002, Suhang Wang |
NeurIPS | 3 |
| 2025 | Examples as the Prompt: A Scalable Approach for Efficient LLM Adaptation in E-CommerceabstractPrompting LLMs offers an efficient way to guide output generation without explicit model training. In the e-commerce domain, prompt-based applications are widely used in query understanding, recommender systems, and customer support. However, adapting LLMs to different tasks often requires extensive prompt engineering by domain experts, along with frequent updates to align with evolving business needs. Additionally, crafting fully unbiased natural language prompts remains a challenge for humans. To address these challenges, we propose a novel framework, Examples as the Prompt (EaP). Specifically, EaP automatically selects the most representative examples to maximize the few-shot capability of LLMs. It is efficient due to its unsupervised example selection and adaptive to potential data distribution shifts. We validate EaP on four real-world production use cases, demonstrating that it achieves comparable or even superior performance comparing to hand-crafted prompts designed by domain experts. Additionally, we introduce EaPlite, which entirely replaces the natural language components of prompts with labeled examples. EaPlite improves LLM inference speed by up to 70% without compromising performance. The online A/B test shows that using EaP and EaPlite for data labeling can bring significant composite revenue gain by 0.06%. Jingying Zeng, Zhenwei Dai, Hui Liu 0033, Samarth Varshney, Zhiji Liu, Chen Luo 0003, Qi He 0002, Xianfeng Tang |
SIGIR | 2 |
| 2024 | Exploring Query Understanding for Amazon Product SearchabstractOnline shopping platforms, such as Amazon, offer services to billions of people worldwide. Unlike web search or other search engines, product search engines have their unique characteristics, primarily featuring short queries which are mostly a combination of product attributes and structured product search space. The uniqueness of product search underscores the crucial importance of the query understanding component. However, there are limited studies focusing on exploring this impact within real-world product search engines. In this work, we aim to bridge this gap by conducting a comprehensive study and sharing our year-long journey investigating how the query understanding service impacts Amazon Product Search. Firstly, we explore how query understanding-based ranking features influence the ranking process. Next, we delve into how the query understanding system contributes to understanding the performance of a ranking model. Building on the insights gained from our study on the evaluation of the query understanding-based ranking model, we propose a query understanding-based multi-task learning framework for ranking. We present our studies and investigations on Amazon Search. Chen Luo 0003, Xianfeng Tang, Hanqing Lu, Yaochen Xie, Hui Liu 0003, Zhenwei Dai, Limeng Cui, Ashutosh Joshi, Sreyashi Nag, Yang Li 0055, Rahul Goutam, Jiliang Tang, Qi He 0002 |
IEEE Big Data | 6 |
| 2023 | Graph Self-supervised Learning via Proximity Distribution MinimizationabstractSelf-supervised learning (SSL) for graphs is an essential problem since graph data are ubiquitous and labeling can be costly. We argue that existing SSL approaches for graphs have two limitations. First, they rely on corruption techniques such as node attribute perturbation and edge dropping to generate graph views for contrastive learning. These unnatural corruption techniques require extensive tuning efforts and provide marginal improvements. Second, the current approaches require the computation of multiple graph views, which is memory and computationally inefficient. These shortcomings of graph SSL call for a corruption-free single-view learning approach, but the strawman approach of using neighboring nodes as positive examples suffers two problems: it ignores the strength of connections between nodes implied by the graph structure on a macro level, and cannot deal with the high noise in real-world graphs. We propose Proximity Divergence Minimization (PDM), a corruption-free single-view graph SSL approach that overcomes these problems by leveraging node proximity to measure connection strength and denoise the graph structure. Through extensive experiments, we show that PDM achieves up to 4.55% absolute improvement in ROC-AUC on graph SSL tasks over state-of-the-art approaches while being more memory efficient. Moreover, PDM even outperforms supervised training on node classification tasks of ogbn-proteins dataset. Our code is publicly available. Tianyi Zhang 0011, Zhenwei Dai, Zhaozhuo Xu, Anshumali Shrivastava |
UAI | 2 |
| 2021 | Learned Bloom Filters in Adversarial Environments: A Malicious URL Detection Use-CaseabstractLearned Bloom Filters (LBFs) have been recently proposed as an alternative to traditional Bloom filters that can reduce the amount of memory needed to achieve a target false positive probability when representing a given set of elements. LBFs rely on Machine Learning models combined with traditional Bloom filters. However, if LBFs are going to be used as an alternative to Bloom filters, their security must be also be considered. In this paper, the security of LBFs is studied for the first time and a vulnerability different from those of traditional Bloom filters is uncovered. In more detail, an attacker can easily create a set of elements that are not in the filter with a much larger false positive probability than the target for which the filter has been designed. The constructed attack set can then be used to for example launch a denial of service attack against the system that uses the LBF. A malicious URL case study is used to illustrate the proposed attacks and show their effectiveness in increasing the false positive probability of LBFs. The dataset under consideration includes nearly 485K URLs where 16.47% of them are malicious URLs. Unfortunately, it seems that mitigating this vulnerability is not straightforward. Pedro Reviriego, José Alberto Hernández 0001, Zhenwei Dai, Anshumali Shrivastava |
HPSR | 3 |
| 2021 | Active Sampling Count Sketch (ASCS) for Online Sparse Estimation of a Trillion Scale Covariance MatrixabstractEstimating and storing the covariance (or correlation) matrix of high-dimensional data is computationally challenging because both memory and computational requirements scale quadratically with the dimension. Fortunately, high-dimensional covariance matrices as observed in text, click-through, meta-genomics datasets, etc are often sparse. In this paper, we consider the problem of efficient sparse estimation of covariance matrices with possibly trillions of entries. The size of the datasets we target requires the algorithm to be online, as more than one pass over the data is prohibitive. In this paper, we propose Active Sampling Count Sketch (ASCS), an online and one-pass sketching algorithm, that recovers the large entries of the covariance matrix accurately. Count Sketch (CS), and other sub-linear compressed sensing algorithms, offer a natural solution to the problem in theory. However, vanilla CS does not work well in practice due to a low signal-to-noise ratio (SNR). At the heart of our approach is a novel active sampling strategy that increases the SNR of classical CS. We demonstrate the practicality of our algorithm with synthetic data and real-world high dimensional datasets. ASCS significantly improves over vanilla CS, demonstrating the merit of our active sampling strategy. Zhenwei Dai, Aditya Desai, Reinhard Heckel, Anshumali Shrivastava |
SIGMOD Conference | 1 |
| 2020 | Adaptive Learned Bloom Filter (Ada-BF): Efficient Utilization of the Classifier with Application to Real-Time Information Filtering on the WebabstractRecent work suggests improving the performance of Bloom filter by incorporating a machine learning model as a binary classifier. However, such learned Bloom filter does not take full advantage of the predicted probability scores. We propose new algorithms that generalize the learned Bloom filter by using the complete spectrum of the score regions. We prove our algorithms have lower false positive rate (FPR) and memory usage compared with the existing approaches to learned Bloom filter. We also demonstrate the improved performance of our algorithms on real-world information filtering tasks over the web. Zhenwei Dai, Anshumali Shrivastava |
NeurIPS | 1 |
| 2019 | Batch effects correction for microbiome data with Dirichlet-multinomial regressionabstractMOTIVATION: Metagenomic sequencing techniques enable quantitative analyses of the microbiome. However, combining the microbial data from these experiments is challenging due to the variations between experiments. The existing methods for correcting batch effects do not consider the interactions between variables-microbial taxa in microbial studies-and the overdispersion of the microbiome data. Therefore, they are not applicable to microbiome data. RESULTS: We develop a new method, Bayesian Dirichlet-multinomial regression meta-analysis (BDMMA), to simultaneously model the batch effects and detect the microbial taxa associated with phenotypes. BDMMA automatically models the dependence among microbial taxa and is robust to the high dimensionality of the microbiome and their association sparsity. Simulation studies and real data analysis show that BDMMA can successfully adjust batch effects and substantially reduce false discoveries in microbial meta-analyses. AVAILABILITY AND IMPLEMENTATION: An R package" BDMMA" for Windows and Linux is available at https://github.com/DAIZHENWEI/BDMMA/BDMMA, and a version for MacOS is provided at https://github.com/DAIZHENWEI/BDMMA/BDMMA_MacOS. SUPPLEMENTARY INFORMATION: Supplementary data are available at Bioinformatics online. Zhenwei Dai, Sunny Hei Wong, Jun Yu 0009 |
Bioinform. | 1 |
| 2019 | Batch effects correction for microbiome data with Dirichlet-multinomial regressionabstractBioinformatics, (2018) doi.org/10.1093/bioinformatics/bty729 In the original article, author Sunny H. Wong’s name was incorrectly spelt as Sunng H. Wong. This has now been corrected. Zhenwei Dai, Sunny Hei Wong, Jun Yu 0009 |
Bioinform. | 1 |