activity
20182023
most citedA Topological Regularizer for Classifiers via Persistent Homology

53 citations · 166 across the 12 of their papers we have counts for

collaborators

12 papers

eess.IV2023★ 3 cited

Learning to Segment from Noisy Annotations: A Spatial Correction Approach

Jiachen Yao, Yikai Zhang, Songzhu Zheng +3

Noisy labels can significantly affect the performance of deep neural networks (DNNs). In medical image segmentation tasks, annotations are error-prone due to the high demand in ann…

cs.LG2022★ 1 cited

A Manifold View of Adversarial Risk

Wenjia Zhang, Yikai Zhang, Xiaoling Hu +3

The adversarial risk of a machine learning model has been widely studied. Most previous works assume that the data lies in the whole ambient space. We propose to take a new angle a…

cs.LG2021★ 2 cited

Topological Detection of Trojaned Neural Networks

Songzhu Zheng, Yikai Zhang, Hubert Wagner +2

Deep neural networks are known to have security issues. One particular threat is the Trojan attack. It occurs when the attackers stealthily manipulate the model's behavior through…

cs.LG2021★ 32 cited

Learning with Feature-Dependent Label Noise: A Progressive Approach

Yikai Zhang, Songzhu Zheng, Pengxiang Wu +2

Label noise is frequently observed in real-world large-scale datasets. The noise is introduced due to a variety of reasons; it is heterogeneous and feature-dependent. Most existing…

cs.LG2021★ 4 cited

Stability of SGD: Tightness Analysis and Improved Bounds

Yikai Zhang, Wenjia Zhang, Sammy Bald +3

Stochastic Gradient Descent (SGD) based methods have been widely used for training large-scale machine learning models that also generalize well in practice. Several explanations h…

cs.LG2021★ 7 cited

Training Federated GANs with Theoretical Guarantees: A Universal Aggregation Approach

Yikai Zhang, Hui Qu, Qi Chang +3

Recently, Generative Adversarial Networks (GANs) have demonstrated their potential in federated learning, i.e., learning a centralized model from data privately hosted by multiple…