Alert button
Picture for Dan Lin

Dan Lin

Alert button

River of No Return: Graph Percolation Embeddings for Efficient Knowledge Graph Reasoning

May 17, 2023
Kai Wang, Siqiang Luo, Dan Lin

Figure 1 for River of No Return: Graph Percolation Embeddings for Efficient Knowledge Graph Reasoning
Figure 2 for River of No Return: Graph Percolation Embeddings for Efficient Knowledge Graph Reasoning
Figure 3 for River of No Return: Graph Percolation Embeddings for Efficient Knowledge Graph Reasoning
Figure 4 for River of No Return: Graph Percolation Embeddings for Efficient Knowledge Graph Reasoning

We study Graph Neural Networks (GNNs)-based embedding techniques for knowledge graph (KG) reasoning. For the first time, we link the path redundancy issue in the state-of-the-art KG reasoning models based on path encoding and message passing to the transformation error in model training, which brings us new theoretical insights into KG reasoning, as well as high efficacy in practice. On the theoretical side, we analyze the entropy of transformation error in KG paths and point out query-specific redundant paths causing entropy increases. These findings guide us to maintain the shortest paths and remove redundant paths for minimized-entropy message passing. To achieve this goal, on the practical side, we propose an efficient Graph Percolation Process motivated by the percolation model in Fluid Mechanics, and design a lightweight GNN-based KG reasoning framework called Graph Percolation Embeddings (GraPE). GraPE outperforms previous state-of-the-art methods in both transductive and inductive reasoning tasks while requiring fewer training parameters and less inference time.

* 9 pages, 2 figures 
Viaarxiv icon

An EEG Channel Selection Framework for Driver Drowsiness Detection via Interpretability Guidance

Apr 26, 2023
Xinliang Zhou, Dan Lin, Ziyu Jia, Jiaping Xiao, Chenyu Liu, Liming Zhai, Yang Liu

Figure 1 for An EEG Channel Selection Framework for Driver Drowsiness Detection via Interpretability Guidance
Figure 2 for An EEG Channel Selection Framework for Driver Drowsiness Detection via Interpretability Guidance
Figure 3 for An EEG Channel Selection Framework for Driver Drowsiness Detection via Interpretability Guidance
Figure 4 for An EEG Channel Selection Framework for Driver Drowsiness Detection via Interpretability Guidance

Drowsy driving has a crucial influence on driving safety, creating an urgent demand for driver drowsiness detection. Electroencephalogram (EEG) signal can accurately reflect the mental fatigue state and thus has been widely studied in drowsiness monitoring. However, the raw EEG data is inherently noisy and redundant, which is neglected by existing works that just use single-channel EEG data or full-head channel EEG data for model training, resulting in limited performance of driver drowsiness detection. In this paper, we are the first to propose an Interpretability-guided Channel Selection (ICS) framework for the driver drowsiness detection task. Specifically, we design a two-stage training strategy to progressively select the key contributing channels with the guidance of interpretability. We first train a teacher network in the first stage using full-head channel EEG data. Then we apply the class activation mapping (CAM) to the trained teacher model to highlight the high-contributing EEG channels and further propose a channel voting scheme to select the top N contributing EEG channels. Finally, we train a student network with the selected channels of EEG data in the second stage for driver drowsiness detection. Experiments are designed on a public dataset, and the results demonstrate that our method is highly applicable and can significantly improve the performance of cross-subject driver drowsiness detection.

Viaarxiv icon

Defend Data Poisoning Attacks on Voice Authentication

Sep 09, 2022
Ke Li, Cameron Baird, Dan Lin

Figure 1 for Defend Data Poisoning Attacks on Voice Authentication
Figure 2 for Defend Data Poisoning Attacks on Voice Authentication
Figure 3 for Defend Data Poisoning Attacks on Voice Authentication
Figure 4 for Defend Data Poisoning Attacks on Voice Authentication

With the advances in deep learning, speaker verification has achieved very high accuracy and is gaining popularity as a type of biometric authentication option in many scenes of our daily life, especially the growing market of web services. Compared to traditional passwords, "vocal passwords" are much more convenient as they relieve people from memorizing different passwords. However, new machine learning attacks are putting these voice authentication systems at risk. Without a strong security guarantee, attackers could access legitimate users' web accounts by fooling the deep neural network (DNN) based voice recognition models. In this paper, we demonstrate an easy-to-implement data poisoning attack to the voice authentication system, which can hardly be captured by existing defense mechanisms. Thus, we propose a more robust defense method, called Guardian, which is a convolutional neural network-based discriminator. The Guardian discriminator integrates a series of novel techniques including bias reduction, input augmentation, and ensemble learning. Our approach is able to distinguish about 95% of attacked accounts from normal accounts, which is much more effective than existing approaches with only 60% accuracy.

Viaarxiv icon

PRSNet: Part Relation and Selection Network for Bone Age Assessment

Sep 05, 2019
Yuanfeng Ji, Hao Chen, Dan Lin, Xiaohua Wu, Di Lin

Figure 1 for PRSNet: Part Relation and Selection Network for Bone Age Assessment
Figure 2 for PRSNet: Part Relation and Selection Network for Bone Age Assessment
Figure 3 for PRSNet: Part Relation and Selection Network for Bone Age Assessment
Figure 4 for PRSNet: Part Relation and Selection Network for Bone Age Assessment

Bone age is one of the most important indicators for assessing bone's maturity, which can help to interpret human's growth development level and potential progress. In the clinical practice, bone age assessment (BAA) of X-ray images requires the joint consideration of the appearance and location information of hand bones. These kinds of information can be effectively captured by the relation of different anatomical parts of hand bone. Recently developed methods differ mostly in how they model the part relation and choose useful parts for BAA. However, these methods neglect the mining of relationship among different parts, which can help to improve the assessment accuracy. In this paper, we propose a novel part relation module, which accurately discovers the underlying concurrency of parts by using multi-scale context information of deep learning feature representation. Furthermore, based on the part relation, we explore a new part selection module, which comprehensively measures the importance of parts and select the top ranking parts for assisting BAA. We jointly train our part relation and selection modules in an end-to-end way, achieving state-of-the-art performance on the public RSNA 2017 Pediatric Bone Age benchmark dataset and outperforming other competitive methods by a significant margin.

Viaarxiv icon

Knowledge Graph Embedding with Entity Neighbors and Deep Memory Network

Aug 11, 2018
Kai Wang, Yu Liu, Xiujuan Xu, Dan Lin

Figure 1 for Knowledge Graph Embedding with Entity Neighbors and Deep Memory Network
Figure 2 for Knowledge Graph Embedding with Entity Neighbors and Deep Memory Network
Figure 3 for Knowledge Graph Embedding with Entity Neighbors and Deep Memory Network
Figure 4 for Knowledge Graph Embedding with Entity Neighbors and Deep Memory Network

Knowledge Graph Embedding (KGE) aims to represent entities and relations of knowledge graph in a low-dimensional continuous vector space. Recent works focus on incorporating structural knowledge with additional information, such as entity descriptions, relation paths and so on. However, common used additional information usually contains plenty of noise, which makes it hard to learn valuable representation. In this paper, we propose a new kind of additional information, called entity neighbors, which contain both semantic and topological features about given entity. We then develop a deep memory network model to encode information from neighbors. Employing a gating mechanism, representations of structure and neighbors are integrated into a joint representation. The experimental results show that our model outperforms existing KGE methods utilizing entity descriptions and achieves state-of-the-art metrics on 4 datasets.

* 9 pages, 4 figures 
Viaarxiv icon