Demonstration venue · read-only. Every page can be browsed; the buttons that would change it are switched off. Create an account to run TaxoReview on your own data. A simulation from public publication records: the people listed were not invited and did not agree to review.
? Tutorial for this page

Zhihua Wei 0001

dblp:55/3674-1 · DBLP profile ↗
← Back
61ranked-venue papers
1as first author
41since 2021 · last 2026
0000-0002-5937-3907ORCID · verified

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

Artificial intelligence and machine learning · 36 · 1 first-author · 25 since 2021Graphics, computer vision, multimedia, augmented reality and games · 20 · 16 since 2021Databases, data management, data science and information retrieval · 15 · 10 since 2021Applied, interdisciplinary, general and emerging computing · 4 · 2 since 2021Human-computer interaction and ubiquitous computing · 1

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
19 papers
Trustworthy machine learning · 25% Vision and language · 16% Language models and text generation · 15%
Databases, data mining, and information retrieval
2 papers
Recommender systems · 75% Graph data management · 25%

Topics — the 30 heaviest of 48, each with the papers that count most for it

TopicWeightPapersLastEvidence papers
Computer vision › 3D vision
point cloud processing
2.132025
A Unified Approach to Interpreting Self-supervised Pre-training Methods for 3D Point Clouds via Interactions · CVPR 2025
Interpretable Rotation-Equivariant Quaternion Neural Networks for 3D Point Cloud Processing · IEEE Trans. Pattern Anal. Mach. Intell. 2024
Verifiability and Predictability: Interpreting Utilities of Network Architectures for Point Cloud Processing · CVPR 2021
Machine learning › Trustworthy machine learning
interpretability
1.932025
Interpreting Arithmetic Reasoning in Large Language Models using Game-Theoretic Interactions · NeurIPS 2025
Interpretable Compositional Convolutional Neural Networks · IJCAI 2021
Verifiability and Predictability: Interpreting Utilities of Network Architectures for Point Cloud Processing · CVPR 2021
Natural language and speech › Language models and text generation
hallucination mitigation
1.822026
Mitigating Action-Relation Hallucinations in LVLMs via Relation-aware Visual Enhancement · ACL (1) 2026
DAMRO: Dive into the Attention Mechanism of LVLM to Reduce Object Hallucination · EMNLP 2024
Computer vision › Face, body and person analysis › person re-identification › robust person re-identification
occluded person re-identification
1.822026
UD-Gaussian: Uncertainty-Driven Gaussian Modeling for Occluded Person Re-Identification · IEEE Trans. Image Process. 2026
Occlusion-Aware Transformer With Second-Order Attention for Person Re-Identification · IEEE Trans. Image Process. 2024
Computer vision › Face, body and person analysis
person re-identification
1.822026
UD-Gaussian: Uncertainty-Driven Gaussian Modeling for Occluded Person Re-Identification · IEEE Trans. Image Process. 2026
Occlusion-Aware Transformer With Second-Order Attention for Person Re-Identification · IEEE Trans. Image Process. 2024
Computer vision › Vision and language
vision-and-language navigation
1.722025
FlexVLN: Flexible Adaptation for Diverse Vision-and-Language Navigation Tasks · IEEE Trans. Multim. 2025
COSMO: Combination of Selective Memorization for Low-Cost Vision-and-Language Navigation · ICCV 2025
Machine learning › Trustworthy machine learning
robustness
1.632024
DAMRO: Dive into the Attention Mechanism of LVLM to Reduce Object Hallucination · EMNLP 2024
ToViLaG: Your Visual-Language Generative Model is Also An Evildoer · EMNLP 2023
Interpretable Rotation-Equivariant Quaternion Neural Networks for 3D Point Cloud Processing · IEEE Trans. Pattern Anal. Mach. Intell. 2024
Machine learning › Trustworthy machine learning
uncertainty estimation
1.222026
UD-Gaussian: Uncertainty-Driven Gaussian Modeling for Occluded Person Re-Identification · IEEE Trans. Image Process. 2026
Occlusion-Aware Transformer With Second-Order Attention for Person Re-Identification · IEEE Trans. Image Process. 2024
Machine learning › Deep learning architectures and training
equivariant neural network
1.222024
Interpretable Rotation-Equivariant Quaternion Neural Networks for 3D Point Cloud Processing · IEEE Trans. Pattern Anal. Mach. Intell. 2024
3D-Rotation-Equivariant Quaternion Neural Networks · ECCV (20) 2020
Machine learning › Learning theory
distribution learning
1.012026
UD-Gaussian: Uncertainty-Driven Gaussian Modeling for Occluded Person Re-Identification · IEEE Trans. Image Process. 2026
Computer vision › Vision and language
vision-language model
1.012026
Mitigating Action-Relation Hallucinations in LVLMs via Relation-aware Visual Enhancement · ACL (1) 2026
Machine learning › Trustworthy machine learning › uncertainty estimation
confidence estimation
0.912025
Improving Prediction Certainty Estimation for Reliable Early Exiting via Null Space Projection · IJCAI 2025
Machine learning › Efficient and distributed learning › adaptive computation
early exit
0.912025
Improving Prediction Certainty Estimation for Reliable Early Exiting via Null Space Projection · IJCAI 2025
Machine learning › Trustworthy machine learning
fairness
0.912025
Raising the Bar: Investigating the Values of Large Language Models via Generative Evolving Testing · ICML 2025
Natural language and speech › Language models and text generation
large language model
0.912025
Raising the Bar: Investigating the Values of Large Language Models via Generative Evolving Testing · ICML 2025
Machine learning › Trustworthy machine learning › interpretability
mechanistic interpretability
0.912025
Interpreting Arithmetic Reasoning in Large Language Models using Game-Theoretic Interactions · NeurIPS 2025
Natural language and speech › Language models and text generation › mathematical reasoning
numerical reasoning
0.912025
Interpreting Arithmetic Reasoning in Large Language Models using Game-Theoretic Interactions · NeurIPS 2025
Computer vision › Vision and language › visual grounding
referring expression comprehension
0.912025
Leveraging Debiased Cross-Modal Attention Maps and Code-Based Reasoning for Zero-Shot Referring Expression Comprehension · ICCV 2025
Machine learning › Trustworthy machine learning
safety evaluation
0.912025
Raising the Bar: Investigating the Values of Large Language Models via Generative Evolving Testing · ICML 2025
Computer vision › 3D vision › point cloud analysis › point cloud learning
self-supervised point cloud pre-training
0.912025
A Unified Approach to Interpreting Self-supervised Pre-training Methods for 3D Point Clouds via Interactions · CVPR 2025
Machine learning › Transfer learning and domain adaptation
zero-shot learning
0.912025
Leveraging Debiased Cross-Modal Attention Maps and Code-Based Reasoning for Zero-Shot Referring Expression Comprehension · ICCV 2025
Natural language and speech › Language models and text generation
alignment
0.812024
On the Essence and Prospect: An Investigation of Alignment Approaches for Big Models · IJCAI 2024
Computer vision › Vision and language
multimodal hallucination
0.812024
DAMRO: Dive into the Attention Mechanism of LVLM to Reduce Object Hallucination · EMNLP 2024
Computer vision › Vision and language › multimodal hallucination
object hallucination
0.812024
DAMRO: Dive into the Attention Mechanism of LVLM to Reduce Object Hallucination · EMNLP 2024
Machine learning › Deep learning architectures and training › attention mechanism › higher-order attention
second-order attention
0.812024
Occlusion-Aware Transformer With Second-Order Attention for Person Re-Identification · IEEE Trans. Image Process. 2024
Machine learning › Deep learning architectures and training
transformer
0.812024
Occlusion-Aware Transformer With Second-Order Attention for Person Re-Identification · IEEE Trans. Image Process. 2024
Natural language and speech › Language models and text generation › large language model safety
detoxification
0.712023
ToViLaG: Your Visual-Language Generative Model is Also An Evildoer · EMNLP 2023
Machine learning › Generative modeling
multimodal generation
0.712023
ToViLaG: Your Visual-Language Generative Model is Also An Evildoer · EMNLP 2023
Computer vision › Vision and language
vision-language generation
0.712023
ToViLaG: Your Visual-Language Generative Model is Also An Evildoer · EMNLP 2023
Machine learning › Representation and self-supervised learning › pre-training › multimodal pretraining
cross-modal pre-training
0.612022
SpeechT5: Unified-Modal Encoder-Decoder Pre-Training for Spoken Language Processing · ACL (1) 2022

Methods — techniques the papers use, named apart from their topics

graph attention · 2.0gaussian modeling · 2.0entropy loss · 2.0discrete haar wavelet transform · 2.0game-theoretic interaction · 1.7cross-modal attention · 1.7quaternion neural network · 1.2relation-aware visual enhancement · 1.0attention enhancement · 1.0joint distribution learning · 0.9interaction decomposition · 0.9generative evolving testing · 0.9adaptive testing · 0.9reinforcement learning · 0.6policy learning · 0.6heterogeneous graph neural network · 0.6graph augmented preference encoder · 0.6
YearPublicationVenuePosition
2026 Mitigating Action-Relation Hallucinations in LVLMs via Relation-aware Visual Enhancement
abstract
Large Vision-Language Models (LVLMs) have achieved remarkable performance on diverse vision-language tasks.However, LVLMs still suffer from hallucinations, generating text that contradicts the visual input.Existing research has primarily focused on mitigating object hallucinations, but often overlooks more complex relation hallucinations, particularly action relations involving interactions between objects.In this study, we empirically observe that the primary cause of action-relation hallucinations in LVLMs is the insufficient attention allocated to visual information.Thus, we propose a framework to locate action-relevant image regions and enhance the LVLM's attention to those regions.Specifically, we define the Action-Relation Sensitivity (ARS) score to identify attention heads that are most sensitive to actionrelation changes, thereby localizing actionrelevant image regions that contain key visual cues.Then, we propose the Relation-aware Visual Enhancement (RVE) method to enhance the LVLM's attention to these action-relevant image regions.Extensive experiments demonstrate that, compared to existing baselines, our method achieves superior performance in mitigating action-relation hallucinations with negligible additional inference cost.Furthermore, it effectively generalizes to spatial-relation hallucinations and object hallucinations.The code is available at https://github.com/ LandHqzx/ARS-RVE.
Zhenxin Qin, Qingzhuo Wang, Ruiyang Qin, Zhihua Wei 0001, Wen Shen 0002
ACL (1)5
2026 Enhancing large language models for knowledge graph question answering via multi-granularity knowledge injection and structured reasoning path-augmented prompting
Chuanyang Gong, Zhihua Wei 0001, Wenhao Tao, Duoqian Miao 0001
Inf. Process. Manag.2
2026 Salient Object Detection Based on Shadowed Sets and Illumination-Guided Network
abstract
Salient object detection (SOD) aims to distinguish salient regions from non-salient ones in an image. In real-world scenarios, factors such as depth variation and surface reflection can interfere with the model's judgment, while illumination uncertainty further intensifies this interference. As a result, the uncertainty in salient boundary detection increases, leading to false or missed detections. To core with uncertainty inherent to the problem, we introduce the concept of shadowed set, which is an effective method to process the uncertainty problem. In this paper, we have designed an illumination-aware feature integration network by conducting dual-input feature integration under the implicit assistance of illumination maps. Firstly, we devised a determination of pixel-level salient area module, which extract illumination maps based on Retinex theory and obtain the main area of salient object based on shadowed set as the implicit feature of illumination. Next, we constructed a dual-modal compression module to solve the problem of feature alignment, which can use the dual-stream structure to process RGB and auxiliary inputs. Finally, multi-stage contextual complementary module can effectively recover fine object edges, and we use the outputs from the last three stages to supervise the training of the entire model. The state originality came from our previous work on illumination maps and shadowed sets, and we creatively combined them with the SOD to process uncertainty of salient area. The experiments demonstrate that our method exhibits excellent performance on multiple RGB-based datasets, at the same time, it also demonstrates unique performance on underwater and challenging scenes.
Hongyun Zhang 0001, Witold Pedrycz, Zhihua Wei 0001, Duoqian Miao 0001
IEEE Trans. Fuzzy Syst.4
2026 UD-Gaussian: Uncertainty-Driven Gaussian Modeling for Occluded Person Re-Identification
abstract
Occluded person re-identification aims to address the identification challenges posed by pedestrians obscured by other individuals or objects. Existing methods often rely on incorporating pose or semantic information to improve model performance under occlusion. However, such information often depends on external models with inevitably cross-domain gaps, whose stability is limited in complex occlusion environments and prone to false results. In this paper, we propose a Transformer-based uncertainty-driven Gaussian model, termed as UD-Gaussian. Firstly, to enrich the detailed features of pedestrian images, a high-frequency enhancement module is introduced. The high-frequency components of the pedestrian image are extracted by Discrete Haar Wavelet Transform, and Top-K high-frequency patches are extracted to construct a graph Laplacian matrix to achieve high-frequency graph attention, which is fused with features learned from self-attention to enhance the high-frequency feature representation. Given the uncertainty in pedestrian feature learning induced by occlusion makes it challenging to obtain reliable and stable pedestrian features, we propose a probability distribution learning module. This module establishes a memory bank to build Gaussian distributions for each pedestrian identity and the entropy is introduced as a loss function to encourage the model to generate more deterministic and relatively independent probability distributions, thereby enhancing the discriminative ability of the model across different pedestrian identities. The high-frequency enhancement module provides a solid foundation for the probability distribution learning module, alleviating uncertainty caused by pedestrian images themselves. Experimental results on occluded and holistic person re-identification datasets demonstrate the superiority of the proposed method.
Yizhang Liu, Hongyun Zhang 0001, Cairong Zhao, Zhihua Wei 0001, Duoqian Miao 0001
IEEE Trans. Image Process.5
2026 Text2Bundle: Towards Personalized Query-based Bundle Generation
abstract
Bundle generation aims to provide a bundle of items for the user, and has been widely studied and applied on online service platforms. Existing bundle generation methods mainly utilized user’s preference from historical interactions in common recommendation paradigm, and ignored the potential textual query which is user’s current explicit intention. There can be a scenario in which a user proactively queries a bundle with some natural language description, the system should be able to generate a bundle that exactly matches the user’s intention through the user’s query and preferences. In this work, we define this user-friendly scenario as Query-based Bundle Generation task and propose a novel framework Text2Bundle that leverages both the user’s short-term interests from the query and the user’s long-term preferences from the historical interactions. Our framework consists of three modules: (1) an intention extractor based on Large Language Model that mines the user’s fine-grained interests from the query; (2) a unified state encoder that learns the current bundle context state and the user’s preferences based on historical interaction and current query; and (3) a bundle generator that generates personalized and complementary bundles using reinforcement learning with specifically designed rewards. We conduct extensive experiments on three real-world datasets and demonstrate the effectiveness of our framework compared with several state-of-the-art methods.
Juntong Hu, Shixuan Zhu, Chuan Cui, Qi Shen 0001, Zhihua Wei 0001
Trans. Recomm. Syst.6
2025 A Unified Approach to Interpreting Self-supervised Pre-training Methods for 3D Point Clouds via Interactions
abstract
Recently, many self-supervised pre-training methods have been proposed to improve the performance of deep neural networks (DNNs) for 3D point clouds processing. However, the common mechanism underlying the effectiveness of different pre-training methods remains unclear. In this paper, we use game-theoretic interactions as a unified approach to explore the common mechanism of pre-training methods. Specifically, we decompose the output score of a DNN into the sum of numerous effects of interactions, with each interaction representing a distinct 3D substructure of the input point cloud. Based on the decomposed interactions, we draw the following conclusions. (1) The common mechanism across different pre-training methods is that they enhance the strength of high-order interactions encoded by DNNs, which represent complex and global 3D structures, while reducing the strength of low-order interactions, which represent simple and local 3D structures. (2) Sufficient pre-training and adequate fine-tuning data for downstream tasks further reinforce the mechanism described above. (3) Pre-training methods carry a potential risk of reducing the transferability of features encoded by DNNs. Inspired by the observed common mechanism, we propose a new method to directly enhance the strength of high-order interactions and reduce the strength of low-order interactions encoded by DNNs, improving performance without the need for pre-training on large-scale datasets. Experiments show that our method achieves performance comparable to traditional pre-training methods.
Jian Ruan, Fanghao Wu, Yuchi Chen, Zhihua Wei 0001, Wen Shen 0002
CVPR5
2025 BRDIA: Bidirectional Reasoning with Dynamic Instruction Adjustment for Multi-hop KGQA
abstract
Multi-hop Knowledge Graph Question Answering (KGQA) involves retrieving entities from knowledge graph (KG) to answer users’ queries. More studies focus on generating instructions by decoding questions to guide path searches on the KG. However, as messages propagate from seed nodes to answer nodes, node embedding predominantly reflects information from forward reasoning paths, without fully considering the complete KG context. Besides, due to relation frequency differences within KG, reasoning may overly rely on common but less informative relations, while ignoring less frequent but more helpful ones. To solve these issues, we propose BRDIA, a GNN-based model that deeply integrates bidirectional reasoning. By introducing a gating mechanism, BRDIA dynamically adjusts bidirectional instructions to build reliable contextual information, better aligning KG with the query. Additionally, we introduce relation frequency to measure relation importance, ensuring that multi-hop reasoning considers both path semantics and prior topological information from the KG. We evaluated BRDIA on two major multi-hop KGQA benchmark datasets. Experimental results demonstrate the effectiveness of our proposed method.
Chuanyang Gong, Zhihua Wei 0001
ICASSP2
2025 Leveraging Debiased Cross-Modal Attention Maps and Code-Based Reasoning for Zero-Shot Referring Expression Comprehension
Wen Shen 0002, Zhihua Wei 0001, Hongyun Zhang 0001
ICCV3
2025 COSMO: Combination of Selective Memorization for Low-Cost Vision-and-Language Navigation
abstract
Vision-and-Language Navigation (VLN) tasks have gained prominence within artificial intelligence research due to their potential application in fields like home assistants. Many contemporary VLN approaches, while based on transformer architectures, have increasingly incorporated additional components such as external knowledge bases or map information to enhance performance. These additions, while boosting performance, also lead to larger models and increased computational costs. In this paper, to achieve both high performance and low computational costs, we propose a novel architecture with the COmbination of Selective MemOrization (COSMO). Specifically, COSMO integrates state-space modules and transformer modules, and incorporates two VLN-customized selective state space modules: the Round Selective Scan (RSS) and the Cross-modal Selective State Space Module (CS3). RSS facilitates comprehensive inter-modal interactions within a single scan, while the CS3 module adapts the selective state space module into a dual-stream architecture, thereby enhancing the acquisition of cross-modal interactions. Experimental validations on three mainstream VLN benchmarks, REVERIE, R2R, and R2R-CE, not only demonstrate competitive navigation performance of our model but also show a significant reduction in computational costs.
Yanyuan Qiao, Qunbo Wang, Zike Yan, Qi Wu 0001, Zhihua Wei 0001, Jing Liu 0001
ICCV6
2025 Raising the Bar: Investigating the Values of Large Language Models via Generative Evolving Testing
abstract
*Warning: Contains harmful model outputs.* Despite significant advancements, the propensity of Large Language Models (LLMs) to generate harmful and unethical content poses critical challenges. Measuring value alignment of LLMs becomes crucial for their regulation and responsible deployment. Although numerous benchmarks have been constructed to assess social bias, toxicity, and ethical issues in LLMs, those static benchmarks suffer from *evaluation chronoeffect*, in which, as models rapidly evolve, existing benchmarks may leak into training data or become saturated, *overestimating* ever-developing LLMs. To tackle this problem, we propose GETA, a novel *generative evolving testing* approach based on adaptive testing methods in measurement theory. Unlike traditional adaptive testing methods that rely on a static test item pool, GETA probes the underlying moral boundaries of LLMs by dynamically generating test items tailored to model capability. GETA co-evolves with LLMs by learning a joint distribution of item difficulty and model value conformity, thus effectively addressing evaluation chronoeffect. We evaluated various popular LLMs with GETA and demonstrated that 1) GETA can dynamically create difficulty-tailored test items and 2) GETA's evaluation results are more consistent with models' performance on unseen OOD and i.i.d. items, laying the groundwork for future evaluation paradigms.
Han Jiang 0007, Xiaoyuan Yi, Zhihua Wei 0001, Ziang Xiao, Xing Xie 0001
ICML3
2025 Improving Prediction Certainty Estimation for Reliable Early Exiting via Null Space Projection
abstract
Early exiting has demonstrated great potential in accelerating the inference of pre-trained language models (PLMs) by enabling easy samples to exit at shallow layers, eliminating the need for executing deeper layers. However, existing early exiting methods primarily rely on class-relevant logits to formulate their exiting signals for estimating prediction certainty, neglecting the detrimental influence of class-irrelevant information in the features on prediction certainty. This leads to an overestimation of prediction certainty, causing premature exiting of samples with incorrect early predictions. To remedy this, we define an NSP score to estimate prediction certainty by considering the proportion of class-irrelevant information in the features. On this basis, we propose a novel early exiting method based on the Certainty-Aware Probability (CAP) score, which integrates insights from both logits and the NSP score to enhance prediction certainty estimation, thus enabling more reliable exiting decisions. The experimental results on the GLUE benchmark show that our method can achieve an average speed-up ratio of 2.19× across all tasks with negligible performance degradation, surpassing the state-of-the-art (SOTA) ConsistentEE by 28%, yielding a better trade-off between task performance and inference efficiency. The code is available at https://github.com/He-Jianing/NSP.git.
Jianing He, Qi Zhang 0020, Duoqian Miao 0001, Kun Yi 0001, Shufeng Hao, Hongyun Zhang 0001, Zhihua Wei 0001
IJCAI7
2025 StyleCSR: Style-controllable Sequential Recommendation with Prompt Tuning
abstract
Controllability is a key factor in building user-trustworthy recommendation systems. However, most existing studies lack consideration for controllability, relying solely on users’ historical interaction data to continuously recommend items that align with their past preferences. This approach may isolate users from the outside world, leading to the "filter bubble" phenomenon. As a result, users may still feel dissatisfied, even when the recommendation accuracy is high. In this work, we propose a user-friendly, style-controllable sequential recommendation framework with prompt tuning (StyleCSR). This framework enables users to provide instructions for adjusting item features, such as popularity and similarity, to control the distribution of recommendation results. Specifically, we map user instructions into prompts and fuse them with user historical behavior sequences. To enhance the controllability of recommendation results while maintaining high recommendation performance, we design an interest alignment module to retain users’ original interests and an instruction discrimination module to emphasize the role of user instructions. We conduct extensive experiments on multiple datasets and various types of pre-trained sequential recommendation models. The results validate that our StyleCSR can be applied to different types of pre-trained models, significantly enhancing the controllability of recommendation results while maintaining high recommendation performance.
Leilei Wen, Qi Shen 0001, Shixuan Zhu, Zhihua Wei 0001, Wen Shen 0002
IJCNN4
2025 Interpreting Arithmetic Reasoning in Large Language Models using Game-Theoretic Interactions
abstract
In recent years, large language models (LLMs) have made significant advancements in arithmetic reasoning. However, the internal mechanism of how LLMs solve arithmetic problems remains unclear. In this paper, we propose explaining arithmetic reasoning in LLMs using game-theoretic interactions. Specifically, we disentangle the output score of the LLM into numerous interactions between the input words. We quantify different types of interactions encoded by LLMs during forward propagation to explore the internal mechanism of LLMs for solving arithmetic problems. We find that (1) the internal mechanism of LLMs for solving simple one-operator arithmetic problems is their capability to encode operand-operator interactions and high-order interactions from input samples. Additionally, we find that LLMs with weak one-operator arithmetic capabilities focus more on background interactions. (2) The internal mechanism of LLMs for solving relatively complex two-operator arithmetic problems is their capability to encode operator interactions and operand interactions from input samples. (3) We explain the task-specific nature of the LoRA method from the perspective of interactions.
Leilei Wen, Liwei Zheng, Zhihua Wei 0001, Wen Shen 0002
NeurIPS5
2025 Binary mask tuning on gradient: Towards multi-data question answering
abstract
Pretrained language models have been widely applied in question-answering tasks. To achieve better generalization on unseen datasets, several previous studies often trained a single model on multiple datasets. However, owing to the noise and distribution differences among various datasets, the model tends to excessively adjust its weights during training on multiple datasets. This deviation from the initial pretraining state results in excellent performance on specific data but overfitting on other datasets. Ultimately, the model loses its ability to generalize to new data. In this paper, from the perspective of imposing constraints on model weights, we propose a novel fine-tuning method, binary mask tuning (BMT). We employ a carefully designed binary mask vector that is closely related to the data distribution to mask the gradient generated by backpropagation , achieving precise alignment of the subspace parameters required to fit the data from a huge parameter space. This approach aimed to enhance the adaptability of the model to the data distribution and improve parameter efficiency via more targeted fine-tuning. Our experiments demonstrate that BMT is not only effective in mitigating the tendency of the model to excessively adjust its weights but also in better capturing cross-dataset regularities and dataset-specific attributes in question-answering tasks across different datasets.
Chuanyang Gong, Zhihua Wei 0001, Duoqian Miao 0001
Knowl. Based Syst.2
2025 Infusing Multi-Hop Medical Knowledge Into Smaller Language Models for Biomedical Question Answering
abstract
MedQA-USMLE is a challenging biomedical question answering (BQA) task, as its questions typically involve multi-hop reasoning. To solve this task, BQA systems should possess not only extensive medical professional knowledge but also strong medical reasoning capabilities. While state-of-the-art larger language models, such as Med-PaLM 2, have overcome this challenge, smaller language models (SLMs) still struggle with it. To bridge this gap, we introduces a multi-hop medical knowledge infusion (MHMKI) procedure to endow SLMs with medical reasoning capabilities. Specifically, we categorize MedQA-USMLE questions into distinct reasoning types, then tailor pre-training instances for each type of questions using the semi-structured information and hyperlinks of Wikipedia articles. To enable SLMs to efficiently capture the multi-hop knowledge contained in these instances, we design a reasoning chain masked language model to further pre-train BERT models. Moreover, we convert the pre-training instances into a composite question answering dataset for intermediate fine-tuning of GPT models. We evaluate MHMKI on six SLMs across five datasets spanning three BQA tasks. The results demonstrate that MHMKI consistently improves SLMs' performance, particularly on tasks requiring substantial medical reasoning. For instance, the accuracy of MedQA-USMLE shows a significant increase of 5.3% on average.
Jing Chen 0043, Zhihua Wei 0001, Wen Shen 0002
IEEE J. Biomed. Health Informatics2
2025 FlexVLN: Flexible Adaptation for Diverse Vision-and-Language Navigation Tasks
Yanyuan Qiao, Qunbo Wang, Longteng Guo, Zhihua Wei 0001, Jing Liu 0001
IEEE Trans. Multim.5
2024 DAMRO: Dive into the Attention Mechanism of LVLM to Reduce Object Hallucination
abstract
Despite the great success of Large Vision-Language Models (LVLMs), they inevitably suffer from hallucination.As we know, both the visual encoder and the Large Language Model (LLM) decoder in LVLMs are Transformerbased, allowing the model to extract visual information and generate text outputs via attention mechanisms.We find that the attention distribution of LLM decoder on image tokens is highly consistent with the visual encoder and both distributions tend to focus on particular background tokens rather than the referred objects in the image.We attribute to the unexpected attention distribution to an inherent flaw in the visual encoder itself, which misguides LLMs to over emphasize the redundant information and generate object hallucination.To address the issue, we propose DAMRO, a novel training-free strategy that Dive into Attention Mechanism of LVLM to Reduce Object Hallucination.Specifically, our approach employs classification token (CLS) of ViT to filter out high-attention outlier tokens scattered in the background and then eliminate their influence during decoding stage.We evaluate our method on LVLMs including LLaVA-1.5, LLaVA-NeXT and Instruct-BLIP, using various benchmarks such as POPE, CHAIR, MME and GPT-4V Aided Evaluation.The results demonstrate that our approach significantly reduces the impact of these outlier tokens, thus effectively alleviating the hallucination of LVLMs.
Tianshi Ming, Xinpeng Wang 0001, Zhihua Wei 0001
EMNLP4
2024 On the Essence and Prospect: An Investigation of Alignment Approaches for Big Models
Xinpeng Wang 0001, Shitong Duan, Xiaoyuan Yi, Jing Yao 0003, Shanlin Zhou, Zhihua Wei 0001, Peng Zhang 0060, Dongkuan Xu, Maosong Sun 0001, Xing Xie 0001
IJCAI6
2024 Subgraph-Based Attention Network for Multi-Hop Question Answering
abstract
Multi-hop question answering (QA) requires the model to integrate multiple evidential fragments scattered in the document to infer the answer. At present, while graph neural network-based (GNN-based) approaches have exhibited promising performance in multi-hop QA tasks, some underlying issues have also emerged. The first is that GNN is insensitive to crucial dependencies between "levels" in the graph. The second is that reasoning messages on the graph require multiple hops to reach the answer entity, and this jumping between nodes induced by reasoning significantly disperses attention. To address these issues, we propose a novel model of Subgraph-based Attention Network (SuAN) for multi-hop QA, which achieves controlled message passing in GNN through subgraph decomposition and constrained reasoning order, more in line with the multi-hop reasoning process. In SuAN, we decompose the graph into a series of subgraphs composed of first-order neighboring nodes based on semantics. Subsequently, we update node representations on these subgraphs according to a pre-defined reasoning order. This approach not only narrows the gap between the human reasoning process and the message passing mechanism of GNN but also enables the full exploitation of the implicit or explicit hierarchical property inherent in the graph structure. The experimental results on HotpotQA demonstrate the effectiveness of the proposed approach and help the baseline model achieve significant improvements.
Chuanyang Gong, Zhihua Wei 0001, Xinpeng Wang 0001, Rui Wang 0073
IJCNN2
2024 SamCap: Energy-based Controllable Image Captioning by Gradient-Based Sampling
abstract
Despite remarkable advances in image captioning, existing models still lack the ability to generate controllable and diverse captions. As a solution, controllable image captioning (CIC) has recently gained attention, with the goal of generating image captions that satisfy the constraints of the given control signals. Current CIC methods have two main limitations: (1) They can only handle one specific control signal and lack the ability to handle combinations of multiple control signals. (2) They depend on costly supervised learning from task-specific data, which becomes impractical with increasing model size. To this end, we propose an energy-based sampling method for controllable image captioning, named SamCap. Specifically, by combining various constraint functions with the log likelihood of the image captioner into an energy function, we can generate captions that satisfy the specified constraints through gradient-based sampling. SamCap provides a learning-free and plug-and-play solution, that can integrate with any existing image captioner without task-specific fine-tuning. Extensive results demonstrate that SamCap not only matches the performance of SOTA signal-specific CIC models for single control signals, but also shows significant advantages in handling combinations of multiple control signals.
Zhihua Wei 0001
ICMR3
2024 Towards Multi-subsession Conversational Recommendation
Qi Shen 0001, Shixuan Zhu, Yiming Zhang 0020, Chuan Cui, Zhihua Wei 0001
PAKDD (5)7
2024 Supplementing domain knowledge to BERT with semi-structured information of documents
Jing Chen 0043, Zhihua Wei 0001, Jiaqi Wang 0018, Rui Wang 0073, Chuanyang Gong, Hongyun Zhang 0001, Duoqian Miao 0001
Expert Syst. Appl.2
2024 Incorporating multi-perspective information into reinforcement learning to address multi-hop knowledge graph question answering
Chuanyang Gong, Zhihua Wei 0001, Rui Wang 0073, Jing Chen 0043, Hongyun Zhang 0001, Duoqian Miao 0001
Expert Syst. Appl.2
2024 Interpretable Rotation-Equivariant Quaternion Neural Networks for 3D Point Cloud Processing
abstract
This study proposes a set of generic rules to revise existing neural networks for 3D point cloud processing to rotation-equivariant quaternion neural networks (REQNNs), in order to make feature representations of neural networks to be rotation-equivariant and permutation-invariant. Rotation equivariance of features means that the feature computed on a rotated input point cloud is the same as applying the same rotation transformation to the feature computed on the original input point cloud. We find that the rotation-equivariance of features is naturally satisfied, if a neural network uses quaternion features. Interestingly, we prove that such a network revision also makes gradients of features in the REQNN to be rotation-equivariant w.r.t. inputs, and the training of the REQNN to be rotation-invariant w.r.t. inputs. Besides, permutation-invariance examines whether the intermediate-layer features are invariant, when we reorder input points. We also evaluate the stability of knowledge representations of REQNNs, and the robustness of REQNNs to adversarial rotation attacks. Experiments have shown that REQNNs outperform traditional neural networks in both terms of classification accuracy and robustness on rotated testing samples.
Wen Shen 0002, Zhihua Wei 0001, Qihan Ren, Shikun Huang, Quanshi Zhang
IEEE Trans. Pattern Anal. Mach. Intell.2
2024 Occlusion-Aware Transformer With Second-Order Attention for Person Re-Identification
abstract
Person re-identification (ReID) typically encounters varying degrees of occlusion in real-world scenarios. While previous methods have addressed this using handcrafted partitions or external cues, they often compromise semantic information or increase network complexity. In this paper, we propose a new method from a novel perspective, termed as OAT. Specifically, we first use a Transformer backbone with multiple class tokens for diverse pedestrian feature learning. Given that the self-attention mechanism in the Transformer solely focuses on low-level feature correlations, neglecting higher-order relations among different body parts or regions. Thus, we propose the Second-Order Attention (SOA) module to capture more comprehensive features. To address computational efficiency, we further derive approximation formulations for implementing second-order attention. Observing that the importance of semantics associated with different class tokens varies due to the uncertainty of the location and size of occlusion, we propose the Entropy Guided Fusion (EGF) module for multiple class tokens. By conducting uncertainty analysis on each class token, higher weights are assigned to those with lower information entropy, while lower weights are assigned to class tokens with higher entropy. The dynamic weight adjustment can mitigate the impact of occlusion-induced uncertainty on feature learning, thereby facilitating the acquisition of discriminative class token representations. Extensive experiments have been conducted on occluded and holistic person re-identification datasets, which demonstrate the effectiveness of our proposed method.
Yizhang Liu, Hongyun Zhang 0001, Cairong Zhao, Zhihua Wei 0001, Duoqian Miao 0001
IEEE Trans. Image Process.5
2024 Multi-Interest Multi-Round Conversational Recommendation System with Fuzzy Feedback Based User Simulator
abstract
Conversational recommendation system (CRS) is able to obtain fine-grained and dynamic user preferences based on interactive dialogue. Previous CRS assumes that the user has a clear target item, which often deviates from the real scenario. The user may have a clear single preference for some attribute types (e.g., brand) of items, while for other attribute types (e.g., color), the user may have multiple preferences or even no clear preferences, which leads to multiple acceptable items under multiple combinations of attribute instances. Furthermore, previous works assume that users would provide clear responses to any questions asked by the system. And, they also assume that users would be dedicated to the target item, that is, user would answer “yes” to the attribute corresponding to the target item and answer “no” to other attributes. However, users’ responses to attributes are not completely dependent on target items, but also influenced by users’ inherent interests. Besides, for some over-specific or equivocal questions, the feedback of user might not be clear (“yes”/“no”) and user might give some fuzzy response like “I don’t know”. To address the aforementioned issues, we first propose a more realistic conversational recommendation learning setting, namely Multi-Interest Multi-round Conversational Recommendation (MIMCR), where users may have multiple interests in attribute instance combinations and accept multiple items with partially overlapped combinations of attribute instances. To effectively cope with MIMCR, we propose a novel learning framework, namely Multiple Choice questions based on Multi-Interest Policy Learning. Moreover, we further propose a more realistic User-centric User Simulator with Fuzzy Feedback (UUSFF), which naturally calibrates the user response with additional fuzzy feedback based on user’s inherent preference. To better match the new scenario UUSFF, we propose a simple but effective adaption method for different backbones. Extensive experimental results on several datasets demonstrate the superiority of our methods for the proposed settings.
Qi Shen 0001, Lingfei Wu 0001, Yiming Zhang 0020, Yitong Pang, Zhihua Wei 0001, Fangli Xu, Bo Long, Jian Pei 0001
Trans. Recomm. Syst.5
2023 ToViLaG: Your Visual-Language Generative Model is Also An Evildoer
abstract
Warning: this paper includes model outputs showing offensive content.Recent large-scale Visual-Language Generative Models (VLGMs) have achieved unprecedented improvement in multimodal image/text generation.However, these models might also generate toxic content, e.g., offensive text and pornography images, raising significant ethical risks.Despite exhaustive studies on toxic degeneration of language models, this problem remains largely unexplored within the context of visual-language generation.This work delves into the propensity for toxicity generation and susceptibility to toxic data across various VLGMs.For this purpose, we built ToViLaG, a dataset comprising 32K co-toxic/mono-toxic text-image pairs and 1K innocuous but evocative text that tends to stimulate toxicity.Furthermore, we propose WInToRe, a novel toxicity metric tailored to visual-language generation, which theoretically reflects different aspects of toxicity considering both input and output.On such a basis, we benchmarked the toxicity of a diverse spectrum of VLGMs and discovered that some models do more evil than expected while some are more vulnerable to infection, underscoring the necessity of VLGMs detoxification.Therefore, we develop an innovative information bottleneckbased detoxification method.Our method reduces toxicity while maintaining acceptable generation quality, providing a promising initial solution to this line of research.* Work done as an intern at MSRA mentored by X. Yi.
Xinpeng Wang 0001, Xiaoyuan Yi, Han Jiang 0007, Shanlin Zhou, Zhihua Wei 0001, Xing Xie 0001
EMNLP5
2023 From Pixels to Explanations: Uncovering the Reasoning Process in Visual Question Answering
abstract
Visual reasoning requires models to construct a reasoning process towards the final decision. Previous studies have used attention maps or textual explanations to illustrate the reasoning process, but both have their limitations. Attention maps can be difficult to read, while textual explanations cannot fully describe the process of reasoning, and both are hard to evaluate quantitatively. This paper proposes a novel pixel-to-explanation reasoning model that employs a user-friendly multimodal rationale to depict the reasoning process. The model dissects the question into subquestions, and constructs reasoning cells to retrieve knowledge from the image and question based on these subquestions. The intermediate outcomes from the reasoning cells are translated into object bounding boxes and classes, with the final output beging classified as a standard VQA answer and translated into a complete answer to summarize the entire reasoning process. All the generated results can be combined to produce a human-readable and informative explanation that can be evaluated quantitatively. Besides the interpretability, we achieved a 4.4% improvement over our baseline model on the GQA dataset and attained new state-of-the-art results on the challenging GQA-OOD dataset.
Jing Liu 0001, Zhihua Wei 0001
MMAsia3
2023 Graph Learning Augmented Heterogeneous Graph Neural Network for Social Recommendation
abstract
Social recommendation based on social network has achieved great success in improving the performance of the recommendation system. Since social network (user-user relations) and user-item interactions are both naturally represented as graph-structured data, Graph Neural Networks (GNNs) have thus been widely applied for social recommendation. Despite the superior performance of existing GNNs-based methods, there are still several severe limitations: (i) Few existing GNNs-based methods have considered a single heterogeneous global graph which takes into account user-user relations, user-item interactions, and item-item similarities simultaneously. That may lead to a lack of complex semantic information and rich topological information when encoding users and items based on GNN. (ii) Furthermore, previous methods tend to overlook the reliability of the original user-user relations which may be noisy and incomplete. (iii) More importantly, the item-item connections established by a few existing methods merely using initial rating attributes or extra attributes (such as category) of items, may be inaccurate or sub-optimal with respect to social recommendation. In order to address these issues, we propose an end-to-end heterogeneous global graph learning framework, namely Graph Learning Augmented Heterogeneous Graph Neural Network (GL-HGNN) for social recommendation. GL-HGNN aims to learn a heterogeneous global graph that makes full use of user-user relations, user-item interactions and item-item similarities in a unified perspective. To this end, we design a Graph Learner (GL) method to learn and optimize user-user and item-item connections separately. Moreover, we employ a Heterogeneous Graph Neural Network (HGNN) to capture the high-order complex semantic relations from our learned heterogeneous global graph. To scale up the computation of graph learning, we further present the Anchor-based Graph Learner (AGL) to reduce computational complexity. Extensive experiments on four real-world datasets demonstrate the effectiveness of our model.
Yiming Zhang 0020, Lingfei Wu 0001, Qi Shen 0001, Yitong Pang, Zhihua Wei 0001, Fangli Xu, Ethan Chang, Bo Long
Trans. Recomm. Syst.5
2022 SpeechT5: Unified-Modal Encoder-Decoder Pre-Training for Spoken Language Processing
abstract
Junyi Ao, Rui Wang, Long Zhou, Chengyi Wang, Shuo Ren, Yu Wu, Shujie Liu, Tom Ko, Qing Li, Yu Zhang, Zhihua Wei, Yao Qian, Jinyu Li, Furu Wei. Proceedings of the 60th Annual Meeting of the Association for Computational Linguistics (Volume 1: Long Papers). 2022.
Junyi Ao, Rui Wang 0073, Chengyi Wang 0002, Shuo Ren 0002, Yu Wu 0012, Shujie Liu 0001, Tom Ko, Qing Li 0001, Yu Zhang 0006, Zhihua Wei 0001, Yao Qian, Jinyu Li 0001, Furu Wei
ACL (1)11
2022 Temporal aware Multi-Interest Graph Neural Network for Session-based Recommendation
Qi Shen 0001, Shixuan Zhu, Yitong Pang, Yiming Zhang 0020, Zhihua Wei 0001
ACML5
2022 CHAE: Fine-Grained Controllable Story Generation with Characters, Actions and Emotions
abstract
Story generation has emerged as an interesting yet challenging NLP task in recent years. Some existing studies aim at generating fluent and coherent stories from keywords and outlines; while others attempt to control the global features of the story, such as emotion, style and topic. However, these works focus on coarse-grained control on the story, neglecting control on the details of the story, which is also crucial for the task. To fill the gap, this paper proposes a model for fine-grained control on the story, which allows the generation of customized stories with characters, corresponding actions and emotions arbitrarily assigned. Extensive experimental results on both automatic and human manual evaluations show the superiority of our method. It has strong controllability to generate stories according to the fine-grained personalized guidance, unveiling the effectiveness of our methodology. Our code is available at https://github.com/victorup/CHAE.
Xinpeng Wang 0001, Han Jiang 0007, Zhihua Wei 0001, Shanlin Zhou
COLING3
2022 Intention Adaptive Graph Neural Network for Category-Aware Session-Based Recommendation
Chuan Cui, Qi Shen 0001, Shixuan Zhu, Yitong Pang, Yiming Zhang 0020, Hanning Gao, Zhihua Wei 0001
DASFAA (2)7
2022 Multi-View Self-Attention Based Transformer for Speaker Recognition
abstract
Initially developed for natural language processing (NLP), Transformer model is now widely used for speech processing tasks such as speaker recognition, due to its powerful sequence modeling capabilities. However, conventional self-attention mechanisms are originally designed for modeling textual sequence without considering the characteristics of speech and speaker modeling. Besides, different Transformer variants for speaker recognition have not been well studied. In this work, we propose a novel multi-view self-attention mechanism and present an empirical study of different Transformer variants with or without the proposed attention mechanism for speaker recognition. Specifically, to balance the capabilities of capturing global dependencies and modeling the locality, we propose a multi-view self-attention mechanism for speaker Transformer, in which different attention heads can attend to different ranges of the receptive field. Furthermore, we introduce and compare five Transformer variants with different network architectures, embedding locations, and pooling methods to learn speaker embeddings. Experimental results on the VoxCeleb1 and VoxCeleb2 datasets show that the proposed multi-view self-attention mechanism achieves improvement in the performance of speaker recognition, and the proposed speaker Transformer network attains excellent results compared with state-of-the-art models.
Rui Wang 0073, Junyi Ao, Shujie Liu 0001, Zhihua Wei 0001, Tom Ko, Qing Li 0001, Yu Zhang 0006
ICASSP5
2022 LightHuBERT: Lightweight and Configurable Speech Representation Learning with Once-for-All Hidden-Unit BERT
abstract
Self-supervised speech representation learning has shown promising results in various speech processing tasks.However, the pre-trained models, e.g., HuBERT, are storage-intensive Transformers, limiting their scope of applications under lowresource settings.To this end, we propose LightHuBERT, a once-for-all Transformer compression framework, to find the desired architectures automatically by pruning structured parameters.More precisely, we create a Transformer-based supernet that is nested with thousands of weight-sharing subnets and design a two-stage distillation strategy to leverage the contextualized latent representations from HuBERT.Experiments on automatic speech recognition (ASR) and the SU-PERB benchmark show the proposed LightHuBERT enables over 10 9 architectures concerning the embedding dimension, attention dimension, head number, feed-forward network ratio, and network depth.LightHuBERT outperforms the original HuBERT on ASR and five SUPERB tasks with the Hu-BERT size, achieves comparable performance to the teacher model in most tasks with a reduction of 29% parameters, and obtains a 3.5× compression ratio in three SUPERB tasks, e.g., automatic speaker verification, keyword spotting, and intent classification, with a slight accuracy loss.The code and pre-trained models are available at https://github.com/mechanicalsea/lighthubert.
Rui Wang 0073, Qibing Bai, Junyi Ao, Zhixiang Xiong, Zhihua Wei 0001, Yu Zhang 0006, Tom Ko, Haizhou Li 0001
INTERSPEECH6
2022 Heterogeneous Global Graph Neural Networks for Personalized Session-based Recommendation
abstract
Predicting the next interaction of a short-term interaction session is a challenging task in session-based recommendation. Almost all existing works rely on item transition patterns, and neglect user historical sessions while modeling user preference, which often leads to non-personalized recommendation. And existing personalized session-based recommenders are limited to sessions of the current user, and ignore the useful item-transition patterns from other user's historical sessions. To address these issues, we propose a novel Heterogeneous Global Graph Neural Networks (HG-GNN) to exploit the item transitions over all sessions in a subtle manner for better inferring user preference from the current and historical sessions. To effectively exploit the item transitions over all sessions from users, our global graph contains item transitions of sessions, user-item interactions and global co-occurrence items. Moreover, to capture user preference from sessions comprehensively, we propose a graph augmented preference encoder to learn the session representation. Specifically, we design a novel heterogeneous graph neural network (HGNN) on heterogeneous global graph to learn long-term user preference and item representations with rich semantics. Based on the HGNN, we propose the Personalized Session Encoder to combine the general user preference and temporal interest of the current session to generate the personalized session representation for recommendation. Extensive experimental results on three real-world datasets show that our model outperforms other state-of-the-art methods.
Yitong Pang, Lingfei Wu 0001, Qi Shen 0001, Yiming Zhang 0020, Zhihua Wei 0001, Fangli Xu, Ethan Chang, Bo Long, Jian Pei 0001
WSDM5
2022 Multiple Choice Questions based Multi-Interest Policy Learning for Conversational Recommendation
abstract
Conversational recommendation system (CRS) is able to obtain fine-grained and dynamic user preferences based on interactive dialogue. Previous CRS assumes that the user has a clear target item, which often deviates from the real scenario, that is for many users who resort to CRS, they might not have a clear idea about what they really like. Specifically, the user may have a clear single preference for some attribute types (e.g. brand) of items, while for other attribute types (e.g. color), the user may have multiple preferences or even no clear preferences, which leads to multiple acceptable attribute instances (e.g. black and red) of one attribute type. Therefore, the users could show their preferences over items under multiple combinations of attribute instances rather than a single item with unique combination of all attribute instances. As a result, we first propose a more realistic conversational recommendation learning setting, namely Multi-Interest Multi-round Conversational Recommendation (MIMCR), where users may have multiple interests in attribute instance combinations and accept multiple items with partially overlapped combinations of attribute instances. To effectively cope with the new CRS learning setting, in this paper, we propose a novel learning framework, namely Multiple Choice questions based Multi-Interest Policy Learning (MCMIPL). In order to obtain user preferences more efficiently, the agent generates multiple choice questions rather than binary yes/no ones on specific attribute instance. Furthermore, we propose a union set strategy to select candidate items instead of existing intersection set strategy in order to overcome over-filtering items during the conversation. Finally, we design a Multi-Interest Policy Learning (MIPL) module, which utilizes captured multiple interests of the user to decide next action, either asking attribute instances or recommending items. Extensive experimental results on four datasets demonstrate the superiority of our method for the proposed MIMCR setting.
Yiming Zhang 0020, Lingfei Wu 0001, Qi Shen 0001, Yitong Pang, Zhihua Wei 0001, Fangli Xu, Bo Long, Jian Pei 0001
WWW5
2022 EfficientTDNN: Efficient Architecture Search for Speaker Recognition
abstract
Convolutional neural networks (CNNs), such as the time-delay neural network (TDNN), have shown their remarkable capability in learning speaker embedding. However, they meanwhile bring a huge computational cost in storage size, processing, and memory. Discovering the specialized CNN that meets a specific constraint requires a substantial effort of human experts. Compared with hand-designed approaches, neural architecture search (NAS) appears as a practical technique in automating the manual architecture design process and has attracted increasing interest in spoken language processing tasks such as speaker recognition. In this paper, we propose EfficientTDNN, an efficient architecture search framework consisting of a TDNN-based supernet and a TDNN-NAS algorithm. The proposed supernet introduces temporal convolution of different ranges of the receptive field and feature aggregation of various resolutions from different layers to TDNN. On top of it, the TDNN-NAS algorithm quickly searches for the desired TDNN architecture via weight-sharing subnets, which surprisingly reduces computation while handling the vast number of devices with various resources requirements. Experimental results on the VoxCeleb dataset show the proposed EfficientTDNN enables approximate$10^{13}$architectures concerning depth, kernel, and width. Considering different computation constraints, it achieves a 2.20% equal error rate (EER) with 204 M multiply-accumulate operations (MACs), 1.41% EER with 571 M MACs as well as 0.94% EER with 1.45 G MACs. Comprehensive investigations suggest that the trained supernet generalizes subnets not sampled during training and obtains a favorable trade-off between accuracy and efficiency.
Rui Wang 0073, Zhihua Wei 0001, Haoran Duan 0001, Shouling Ji, Yang Long 0001, Zhen Hong
IEEE ACM Trans. Audio Speech Lang. Process.2
2022 Improved Instance Discrimination and Feature Compactness for End-to-End Person Search
abstract
Person search aims to locate and retrieve specific pedestrians in scene images, including two subtasks, pedestrian detection and person re-identification. Recently, triplet loss has been widely used in person re-identification, which effectively improves the pedestrian features embedding and achieves superior performance. However, forming triplet in the person search is not an easy task. Most of the existing end-to-end person search methods are based on Faster R-CNN. The training process of person re-identification part is affected by the detector. It is difficult to form pedestrian triplets within a limited batch size. Also, there are many pedestrian identities in the person search dataset, but each pedestrian identity only has a few samples. It is difficult to learn a robust pedestrian feature representation for person search. To resolve the problem discussed above, a novel Feature Compactness (FC) Loss for the person search is designed, which efficiently improves the inter-class discrimination and intra-class compactness of pedestrian features embedding without the need for positive or negative pairs. Besides, we propose a pedestrian attention module (PAM) to help the network focuses more on pedestrian information and suppresses irrelevant background information. Our method achieves comparable performance on two benchmarks, CUHK-SYSU and PRW, and achieves 91.96% of mAP and 93.34% of rank1 accuracy on CUHK-SYSU.
Shaowei Hou, Cairong Zhao, Jun Wu 0006, Zhihua Wei 0001, Duoqian Miao 0001
IEEE Trans. Circuits Syst. Video Technol.5
2021 Verifiability and Predictability: Interpreting Utilities of Network Architectures for Point Cloud Processing
abstract
In this paper, we diagnose deep neural networks for 3D point cloud processing to explore utilities of different intermediate-layer network architectures. We propose a number of hypotheses on the effects of specific intermediate-layer network architectures on the representation capacity of DNNs. In order to prove the hypotheses, we design five metrics to diagnose various types of DNNs from the following perspectives, information discarding, information concentration, rotation robustness, adversarial robustness, and neighborhood inconsistency. We conduct comparative studies based on such metrics to verify the hypotheses. We further use the verified hypotheses to revise intermediate-layer architectures of existing DNNs and improve their utilities. Experiments demonstrate the effectiveness of our method. The code will be released when this paper is accepted.
Wen Shen 0002, Zhihua Wei 0001, Shikun Huang, Panyue Chen, Quanshi Zhang
CVPR2
2021 Interpretable Compositional Convolutional Neural Networks
abstract
This paper proposes a method to modify a traditional convolutional neural network (CNN) into an interpretable compositional CNN, in order to learn filters that encode meaningful visual patterns in intermediate convolutional layers. In a compositional CNN, each filter is supposed to consistently represent a specific compositional object part or image region with a clear meaning. The compositional CNN learns from image labels for classification without any annotations of parts or regions for supervision. Our method can be broadly applied to different types of CNNs. Experiments have demonstrated the effectiveness of our method. The code will be released when the paper is accepted.
Wen Shen 0002, Zhihua Wei 0001, Shikun Huang, Quanshi Zhang
IJCAI2
2020 3D-Rotation-Equivariant Quaternion Neural Networks
Wen Shen 0002, Shikun Huang, Zhihua Wei 0001, Quanshi Zhang
ECCV (20)4
2020 Time-Aware Attentive Neural Network for News Recommendation with Long- and Short-Term User Representation
Yitong Pang, Yiming Zhang 0020, Jianing Tong, Zhihua Wei 0001
KSEM (2)4
2020 Non-numerical nearest neighbor classifiers with value-object hierarchical embedding
Sheng Luo 0006, Duoqian Miao 0001, Zhihua Wei 0001
Expert Syst. Appl.4
2020 Improved general attribute reduction algorithms
Baizhen Li, Zhihua Wei 0001, Duoqian Miao 0001, Nan Zhang 0041, Wen Shen 0002, Hongyun Zhang 0001
Inf. Sci.2
2020 RepeatPadding: Balancing words and sentence length for language comprehension in visual question answering
Yu Long 0003, Pengjie Tang, Zhihua Wei 0001, Jinjing Gu, Hanli Wang
Inf. Sci.3
2020 Three-way decisions based blocking reduction models in hierarchical classification
Wen Shen 0002, Zhihua Wei 0001, Qianwen Li, Hongyun Zhang 0001, Duoqian Miao 0001
Inf. Sci.2
2019 A Dustbin Category Based Feedback Incremental Learning Strategy for Hierarchical Image Classification
Wen Shen 0002, Qianwen Li, Zhihua Wei 0001
PRCV (1)4
2019 Multi-granularity principal curves extraction based on improved spectral clustering of complex distribution data
Hongyun Zhang 0001, Zhihua Wei 0001
Int. J. Approx. Reason.4
2019 A self-adaptive cascade ConvNets model based on label relation mining
Zhihua Wei 0001, Wen Shen 0002, Cairong Zhao, Duoqian Miao 0001
Neurocomputing1
2019 QRKISS: A Two-Stage Metric Learning via QR-Decomposition and KISS for Person Re-Identification
Cairong Zhao, Yipeng Chen, Zhihua Wei 0001, Duoqian Miao 0001, Xinjian Gu
Neural Process. Lett.3
2019 Multilevel triplet deep learning model for person re-identification
Cairong Zhao, Zhihua Wei 0001, Yipeng Chen, Duoqian Miao 0001
Pattern Recognit. Lett.3
2019 Long-Term Traffic Speed Prediction Based on Multiscale Spatio-Temporal Feature Learning Network
abstract
Speed plays a significant role in evaluating the evolution of traffic status, and predicting speed is one of the fundamental tasks for the intelligent transportation system. There exists a large number of works on speed forecast; however, the problem of long-term prediction for the next day is still not well addressed. In this paper, we propose a multiscale spatio-temporal feature learning network (MSTFLN) as the model to handle the challenging task of long-term traffic speed prediction for elevated highways. Raw traffic speed data collected from loop detectors every 5 min are transformed into spatial-temporal matrices; each matrix represents the one-day speed information, rows of the matrix indicate the numbers of loop detectors, and time intervals are denoted by columns. To predict the traffic speed of a certain day, nine speed matrices of three historical days with three different time scales are served as the input of MSTFLN. The proposed MSTFLN model consists of convolutional long short-term memories and convolutional neural networks. Experiments are evaluated using the data of three main elevated highways in Shanghai, China. The presented results demonstrate that our approach outperforms the state-of-the-art work and it can effectively predict the long-term speed information.
Di Zang, Jiawei Ling, Zhihua Wei 0001, Keshuang Tang, Jiujun Cheng
IEEE Trans. Intell. Transp. Syst.3
2016 Fusion of multiple channel features for person re-identification
Xuekuan Wang, Cairong Zhao, Duoqian Miao 0001, Zhihua Wei 0001, Renxian Zhang, Tingfei Ye
Neurocomputing4
2015 Lane Detection Method Based on Improved RANSAC Algorithm
abstract
Lane detection based on computer vision is a key technology of Automatic Drive System for intelligent vehicles. In this paper, we propose a real-time and efficient lane detection algorithm that can detect lanes appearing in urban streets and highway roads under complex background. In order to enhance lane boundary information and to be suitable for various light conditions, we adopt canny algorithm for edge detection to get good feature points. We use the generalized curve lane parameter model, which can describe both straight and curved lanes. We propose an improved random sample consensus (RANSAC) algorithm combined with the least squares technique to estimate lane model parameters based on feature extraction. Experiments are conducted on both real road lane videos captured by Tongji University and Caltech Lane Datasets. The experimental results show that our algorithm is can meet the real time requirement and fit lane boundaries well in various challenging road conditions.
Zhihua Wei 0001, Duoqian Miao 0001
ISADS2
2015 Multiple Granular Analysis of TCM Data with Applications on Diagnosis of Hepatitis B
abstract
The objectiveness of Traditional Chinese Medicine (TCM) limits its further development and generalization. Big data provide the golden opportunity for TCM quantization. The main purpose of this paper is to build a bridge between data analysis and clinical experience and provide experimental support for TCM experience. Taking the Hepatitis B disease data as experimental subject, we propose a framework for mining latent relations between features and disease categories based on Granular Computing theory. That is, kmeans clustering and correlation analysis is adopted to analyze the intra-relationship of disease stages and relationship between clinical symptoms and stages respectively. Algorithm based on Latent Dirichlet Allocation model is proposed to mining the mapping relationships of the three layers: clinical symptoms, stages of Hepatitis B and their middle layer syndromes. Experimental results indicate that the results of the data analysis are consistent with the clinical experience of TCM. It is proved that the diagnose based on syndromes is the scientific results of manually mining large amount of history data. Our study is a useful attempt that using data mining techniques to make TCM quantifiable and objective.
Wen Shen 0002, Zhihua Wei 0001, Yunyi Li
SMC2
2014 A Framework of Video Coding for Compressing Near-Duplicate Videos
Hanli Wang, Yu-Gang Jiang 0001, Zhihua Wei 0001
MMM (1)4
2014 Early detection of all-zero 4×4 blocks in High Efficiency Video Coding
Hanli Wang, Weiyao Lin, Sam Kwong, Oscar C. Au, Jun Wu 0006, Zhihua Wei 0001
J. Vis. Commun. Image Represent.7
2014 Graph embedding discriminant analysis for face recognition
Cairong Zhao, Zhihui Lai 0001, Duoqian Miao 0001, Zhihua Wei 0001, Caihui Liu
Neural Comput. Appl.4
2014 From Principal Curves to Granular Principal Curves
abstract
Principal curves arising as an essential construct in dimensionality reduction and data analysis have recently attracted much attention from theoretical as well as practical perspective. In many real-world situations, however, the efficiency of existing principal curves algorithms is often arguable, in particular when dealing with massive data owing to the associated high computational complexity. A certain drawback of these constructs stems from the fact that in several applications principal curves cannot fully capture some essential problem-oriented facets of the data dealing with width, aspect ratio, width change, etc. Information granulation is a powerful tool supporting processing and interpreting massive data. In this paper, invoking the underlying ideas of information granulation, we propose a granular principal curves approach, regarded as an extension of principal curves algorithms, to improve efficiency and achieve a sound accuracy-efficiency tradeoff. First, large amounts of numerical data are granulated into C intervals-information granules developed with the use of fuzzy C-means clustering and the two criteria of information granulation, which significantly reduce the amount of data to be processed at the later phase of the overall design. Granular principal curves are then constructed by determining the upper and the lower bounds of the interval data. Finally, we develop an objective function using the criteria of information confidence and specificity to evaluate the granular output formed by the principal curves. We also optimize the granular principal curves by adjusting the level of information granularity (the number of clusters), which is realized with the aid of the particle swarm optimization. A number of numeric studies completed for synthetic and real-world datasets provide a useful quantifiable insight into the effectiveness of the proposed algorithm.
Hongyun Zhang 0001, Witold Pedrycz, Duoqian Miao 0001, Zhihua Wei 0001
IEEE Trans. Cybern.4
2012 Document-Level Sentiment Classification Based on Behavior-Knowledge Space Method
Duoqian Miao 0001, Zhihua Wei 0001
ADMA3