Wenkai Shi

dblp:252/1153 · DBLP profile ↗
← Back
12ranked-venue papers
2as first author
11since 2021 · last 2026
—ORCID · conflict

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

Artificial intelligence and machine learning · 6 · 2 first-author · 6 since 2021Graphics, computer vision, multimedia, augmented reality and games · 6 · 1 first-author · 6 since 2021Systems, architecture and hardware · 1 · 1 since 2021Computer networks · 1Software engineering, systems software and programming languages · 1 · 1 since 2021Databases, data management, data science and information retrieval · 1 · 1 since 2021
YearPublicationVenuePosition
2026 Enterprise efficiency analysis based on explainable artificial intelligence: From predictive algorithms to mechanisms
Hailin Li, Hufeng Li, Wenkai Shi, Yen-Chun Jim Wu
Inf. Process. Manag.3
2025 Unleashing the Potential of Model Bias for Generalized Category Discovery
abstract
Generalized Category Discovery is a significant and complex task that aims to identify both known and undefined novel categories from a set of unlabeled data, leveraging another labeled dataset containing only known categories. The primary challenges stem from model bias induced by pre-training on only known categories and the lack of precise supervision for novel ones, leading to category bias towards known categories and category confusion among different novel categories, which hinders models' ability to identify novel categories effectively. To address these challenges, we propose a novel framework named Self-Debiasing Calibration (SDC). Unlike prior methods that regard model bias towards known categories as an obstacle to novel category identification, SDC provides a novel insight into unleashing the potential of the bias to facilitate novel category learning. Specifically, we utilize the biased pre-trained model to guide the subsequent learning process on unlabeled data. The output of the biased model serves two key purposes. First, it provides an accurate modeling of category bias, which can be utilized to measure the degree of bias and debias the output of the current training model. Second, it offers valuable insights for distinguishing different novel categories by transferring knowledge between similar categories. Based on these insights, SDC dynamically adjusts the output logits of the current training model using the output of the biased model. This approach produces less biased logits to effectively address the issue of category bias towards known categories, and generates more accurate pseudo labels for unlabeled data, thereby mitigating category confusion for novel categories. Experiments on three benchmark datasets show that SDC outperforms SOTA methods, especially in the identification of novel categories.
Wenbin An, Haonan Lin, Jiahao Nie 0002, Feng Tian 0002, Wenkai Shi, Yaqiang Wu, Qianying Wang 0002, Ping Chen 0001
AAAI5
2025 OS2G: A High-Performance DPU Offloading Architecture for GPU-based Deep Learning with Object Storage
abstract
Object storage is increasingly attractive for deep learning (DL) applications due to its cost-effectiveness and high scalability. However, it exacerbates CPU burdens in DL clusters due to intensive object storage processing and multiple data movements. Data processing unit (DPU) offloading is a promising solution, but naively offloading the existing object storage client leads to severe performance degradation. Besides, only offloading the object storage client still involves redundant data movements, as data must first transfer from the DPU to the host and then from the host to the GPU, which continues to consume valuable host resources.
Zhen Jin 0008, Yiquan Chen, Mingxu Liang, Guoju Fang, Keyao Zhang, Jiexiong Xu, Wenhai Lin, Yiquan Lin, Shushu Zhao, Wenkai Shi, Zhenhua He, Shishun Cai, Wenzhi Chen
ASPLOS (2)12
2024 Transfer and Alignment Network for Generalized Category Discovery
abstract
Generalized Category Discovery (GCD) is a crucial real-world task that aims to recognize both known and novel categories from an unlabeled dataset by leveraging another labeled dataset with only known categories. Despite the improved performance on known categories, current methods perform poorly on novel categories. We attribute the poor performance to two reasons: biased knowledge transfer between labeled and unlabeled data and noisy representation learning on the unlabeled data. The former leads to unreliable estimation of learning targets for novel categories and the latter hinders models from learning discriminative features. To mitigate these two issues, we propose a Transfer and Alignment Network (TAN), which incorporates two knowledge transfer mechanisms to calibrate the biased knowledge and two feature alignment mechanisms to learn discriminative features. Specifically, we model different categories with prototypes and transfer the prototypes in labeled data to correct model bias towards known categories. On the one hand, we pull instances with known categories in unlabeled data closer to these prototypes to form more compact clusters and avoid boundary overlap between known and novel categories. On the other hand, we use these prototypes to calibrate noisy prototypes estimated from unlabeled data based on category similarities, which allows for more accurate estimation of prototypes for novel categories that can be used as reliable learning targets later. After knowledge transfer, we further propose two feature alignment mechanisms to acquire both instance- and category-level knowledge from unlabeled data by aligning instance features with both augmented features and the calibrated prototypes, which can boost model performance on both known and novel categories with less noise. Experiments on three benchmark datasets show that our model outperforms SOTA methods, especially on novel categories. Theoretical analysis is provided for an in-depth understanding of our model in general. Our code and data are available at https://github.com/Lackel/TAN.
Wenbin An, Feng Tian 0002, Wenkai Shi, Yan Chen 0031, Yaqiang Wu, Qianying Wang 0002, Ping Chen 0001
AAAI3
2024 A Unified Knowledge Transfer Network for Generalized Category Discovery
abstract
Generalized Category Discovery (GCD) aims to recognize both known and novel categories in an unlabeled dataset by leveraging another labeled dataset with only known categories. Without considering knowledge transfer from known to novel categories, current methods usually perform poorly on novel categories due to the lack of corresponding supervision. To mitigate this issue, we propose a unified Knowledge Transfer Network (KTN), which solves two obstacles to knowledge transfer in GCD. First, the mixture of known and novel categories in unlabeled data makes it difficult to identify transfer candidates (i.e., samples with novel categories). For this, we propose an entropy-based method that leverages knowledge in the pre-trained classifier to differentiate known and novel categories without requiring extra data or parameters. Second, the lack of prior knowledge of novel categories presents challenges in quantifying semantic relationships between categories to decide the transfer weights. For this, we model different categories with prototypes and treat their similarities as transfer weights to measure the semantic similarities between categories. On the basis of two treatments, we transfer knowledge from known to novel categories by conducting pre-adjustment of logits and post-adjustment of labels for transfer candidates based on the transfer weights between different categories. With the weighted adjustment, KTN can generate more accurate pseudo-labels for unlabeled data, which helps to learn more discriminative features and boost model performance on novel categories. Extensive experiments show that our method outperforms state-of-the-art models on all evaluation metrics across multiple benchmark datasets. Furthermore, different from previous clustering-based methods that can only work offline with abundant data, KTN can be deployed online conveniently with faster inference speed. Code and data are available at https://github.com/yibai-shi/KTN.
Wenkai Shi, Wenbin An, Feng Tian 0002, Yan Chen 0031, Yaqiang Wu, Qianying Wang 0002, Ping Chen 0001
AAAI1
2024 Point-GCC: Universal Self-supervised 3D Scene Pre-training via Geometry-Color Contrast
Guofan Fan, Zekun Qi, Wenkai Shi, Kaisheng Ma
ACM Multimedia3
2024 DOWN: Dynamic Order Weighted Network for Fine-grained Category Discovery
Wenbin An, Feng Tian 0002, Wenkai Shi, Haonan Lin, Yaqiang Wu, Mingxiang Cai, Luyan Wang, Hua Wen, Ping Chen 0001
Knowl. Based Syst.3
2023 DNA: Denoised Neighborhood Aggregation for Fine-grained Category Discovery
abstract
Discovering fine-grained categories from coarsely labeled data is a practical and challenging task, which can bridge the gap between the demand for fine-grained analysis and the high annotation cost.Previous works mainly focus on instance-level discrimination to learn low-level features, but ignore semantic similarities between data, which may prevent these models learning compact cluster representations.In this paper, we propose Denoised Neighborhood Aggregation (DNA), a self-supervised framework that encodes semantic structures of data into the embedding space.Specifically, we retrieve k-nearest neighbors of a query as its positive keys to capture semantic similarities between data and then aggregate information from the neighbors to learn compact cluster representations, which can make fine-grained categories more separatable.However, the retrieved neighbors can be noisy and contain many false-positive keys, which can degrade the quality of learned embeddings.To cope with this challenge, we propose three principles to filter out these false neighbors for better representation learning.Furthermore, we theoretically justify that the learning objective of our framework is equivalent to a clustering loss, which can capture semantic similarities between data to form compact fine-grained clusters.Extensive experiments on three benchmark datasets show that our method can retrieve more accurate neighbors (21.31% accuracy improvement) and outperform state-of-the-art models by a large margin (average 9.96% improvement on three metrics).Our code and data are available at https://github.com/Lackel/DNA.
Wenbin An, Feng Tian 0002, Wenkai Shi, Yan Chen 0031, Qianying Wang 0002, Ping Chen 0001
EMNLP3
2023 A Diffusion Weighted Graph Framework for New Intent Discovery
abstract
New Intent Discovery (NID) aims to recognize both new and known intents from unlabeled data with the aid of limited labeled data containing only known intents.Without considering structure relationships between samples, previous methods generate noisy supervisory signals which cannot strike a balance between quantity and quality, hindering the formation of new intent clusters and effective transfer of the pre-training knowledge.To mitigate this limitation, we propose a novel Diffusion Weighted Graph Framework (DWGF) to capture both semantic similarities and structure relationships inherent in data, enabling more sufficient and reliable supervisory signals.Specifically, for each sample, we diffuse neighborhood relationships along semantic paths guided by the nearest neighbors for multiple hops to characterize its local structure discriminately.Then, we sample its positive keys and weigh them based on semantic similarities and local structures for contrastive learning.During inference, we further propose Graph Smoothing Filter (GSF) to explicitly utilize the structure relationships to filter high-frequency noise embodied in semantically ambiguous samples on the cluster boundary.Extensive experiments show that our method outperforms state-of-the-art models on all evaluation metrics across multiple benchmark datasets.
Wenkai Shi, Wenbin An, Feng Tian 0002, Qianying Wang 0002, Ping Chen 0001
EMNLP1
2021 Enhancement of ridge-valley features in point cloud based on position and normal guidance
Jianhui Nie, Zhaochen Zhang, Ye Liu 0005, Hao Gao 0005, Feng Xu 0005, Wenkai Shi
Comput. Graph.6
2021 Bas-relief generation from point clouds based on normal space compression with real-time adjustment on CPU
Jianhui Nie, Wenkai Shi, Ye Liu 0005, Hao Gao 0005, Feng Xu 0005, Zhaochen Zhang
Graph. Model.2
2019 Nonuniform Node Distribution using Adaptive Poisson Disk for Wireless Sensor Networks
abstract
In this paper, we investigate a nonuniform node distribution strategy to mitigate the energy hole problem in wireless sensor networks (WSNs). Firstly, based on the analysis of energy consumption, we deduce a novel continuous node density function. Secondly, with the transformation of the density function, we obtain the disk radius function. And then, based on the disk radius function, we propose a novel nonuniform node distribution using adaptive Poisson disk (NDAPD). Our strategy can make the node density vary continuously in the network and mitigate the energy hole problem effectively. Finally, a routing algorithm tailored is presented for the proposed nonuniform node distribution. Compared with other well-known nonuniform node distribution strategies, NDAPD can reduce the number of network nodes effectively, improve the utilization of network energy by 3%-5% and increase the data transmission rate to 100% in most cases. Like other strategies, our strategy can be applied to most scenarios.
Xiang Ying, Mei Yu 0004, Wenkai Shi, Jianrong Wang
WCNC5