Search Results for author: Heng Huang

Found 215 papers, 47 papers with code

Adversarial Nonnegative Matrix Factorization

no code implementations ICML 2020 lei luo, yanfu Zhang, Heng Huang

Nonnegative Matrix Factorization (NMF) has become an increasingly important research topic in machine learning.

Bilevel Optimization

Can Stochastic Zeroth-Order Frank-Wolfe Method Converge Faster for Non-Convex Problems?

no code implementations ICML 2020 Hongchang Gao, Heng Huang

To address the problem of lacking gradient in many applications, we propose two new stochastic zeroth-order Frank-Wolfe algorithms and theoretically proved that they have a faster convergence rate than existing methods for non-convex problems.

Sparse Shrunk Additive Models

no code implementations ICML 2020 Hong Chen, Guodong Liu, Heng Huang

Meanwhile, in these feature selection models, the interactions between features are often ignored or just discussed under prior structure information.

Additive models feature selection +1

Fast OSCAR and OWL with Safe Screening Rules

no code implementations ICML 2020 Runxue Bao, Bin Gu, Heng Huang

Ordered Weight $L_{1}$-Norms (OWL) is a new family of regularizers for high-dimensional sparse regression.

regression

AutoTrust: Benchmarking Trustworthiness in Large Vision Language Models for Autonomous Driving

1 code implementation19 Dec 2024 Shuo Xing, Hongyuan Hua, Xiangbo Gao, Shenzhe Zhu, Renjie Li, Kexin Tian, Xiaopeng Li, Heng Huang, Tianbao Yang, Zhangyang Wang, Yang Zhou, Huaxiu Yao, Zhengzhong Tu

Our findings call for immediate and decisive action to address the trustworthiness of DriveVLMs -- an issue of critical importance to public safety and the welfare of all citizens relying on autonomous transportation systems.

Autonomous Driving Benchmarking +4

SleeperMark: Towards Robust Watermark against Fine-Tuning Text-to-image Diffusion Models

no code implementations6 Dec 2024 Zilan Wang, Junfeng Guo, Jiacheng Zhu, Yiming Li, Heng Huang, Muhao Chen, Zhengzhong Tu

Recent advances in large-scale text-to-image (T2I) diffusion models have enabled a variety of downstream applications, including style customization, subject-driven personalization, and conditional generation.

Cell as Point: One-Stage Framework for Efficient Cell Tracking

no code implementations22 Nov 2024 Yaxuan Song, Jianan Fan, Heng Huang, Mei Chen, Weidong Cai

Cellular activities are dynamic and intricate, playing a crucial role in advancing diagnostic and therapeutic techniques, yet they often require substantial resources for accurate tracking.

Cell Tracking Segmentation

APDDv2: Aesthetics of Paintings and Drawings Dataset with Artist Labeled Scores and Comments

1 code implementation13 Nov 2024 Xin Jin, Qianqian Qiao, Yi Lu, Huaye Wang, Heng Huang, Shan Gao, Jianfei Liu, Rui Li

Datasets play a pivotal role in training visual models, facilitating the development of abstract understandings of visual features through diverse image samples and multidimensional attributes.

Provably Faster Algorithms for Bilevel Optimization via Without-Replacement Sampling

no code implementations7 Nov 2024 Junyi Li, Heng Huang

Bilevel Optimization has experienced significant advancements recently with the introduction of new efficient algorithms.

Bilevel Optimization

Fast and scalable Wasserstein-1 neural optimal transport solver for single-cell perturbation prediction

1 code implementation1 Nov 2024 Yanshuo Chen, Zhengmian Hu, Wei Chen, Heng Huang

Our experiments demonstrate that the proposed $W_1$ neural optimal transport solver can mimic the $W_2$ OT solvers in finding a unique and ``monotonic" map on 2D datasets.

Inevitable Trade-off between Watermark Strength and Speculative Sampling Efficiency for Language Models

no code implementations27 Oct 2024 Zhengmian Hu, Heng Huang

Large language models are probabilistic models, and the process of generating content is essentially sampling from the output distribution of the language model.

Language Modeling Language Modelling

A Watermark for Order-Agnostic Language Models

no code implementations17 Oct 2024 Ruibo Chen, Yihan Wu, Yanshuo Chen, Chenxi Liu, Junfeng Guo, Heng Huang

Correspondingly, we propose a statistical pattern-based detection algorithm that recovers the key sequence during detection and conducts statistical tests based on the count of high-frequency patterns.

De-mark: Watermark Removal in Large Language Models

no code implementations17 Oct 2024 Ruibo Chen, Yihan Wu, Junfeng Guo, Heng Huang

Watermarking techniques offer a promising way to identify machine-generated content via embedding covert information into the contents generated from language models (LMs).

OmnixR: Evaluating Omni-modality Language Models on Reasoning across Modalities

no code implementations16 Oct 2024 Lichang Chen, Hexiang Hu, Mingda Zhang, YiWen Chen, Zifeng Wang, Yandong Li, Pranav Shyam, Tianyi Zhou, Heng Huang, Ming-Hsuan Yang, Boqing Gong

To address this, OmnixR offers two evaluation variants: (1)synthetic subset: a synthetic dataset generated automatically by translating text into multiple modalities--audio, images, video, and hybrids (Omnify).

Cross-Modal Few-Shot Learning: a Generative Transfer Learning Framework

no code implementations14 Oct 2024 Zhengwei Yang, Yuke Li, Qiang Sun, Basura Fernando, Heng Huang, Zheng Wang

Most existing studies on few-shot learning focus on unimodal settings, where models are trained to generalize on unseen data using only a small number of labeled examples from the same modality.

Few-Shot Learning Transfer Learning

LLaVA-Critic: Learning to Evaluate Multimodal Models

no code implementations3 Oct 2024 Tianyi Xiong, Xiyao Wang, Dong Guo, Qinghao Ye, Haoqi Fan, Quanquan Gu, Heng Huang, Chunyuan Li

We introduce LLaVA-Critic, the first open-source large multimodal model (LMM) designed as a generalist evaluator to assess performance across a wide range of multimodal tasks.

Instruction Following

Obtaining Lower Query Complexities through Lightweight Zeroth-Order Proximal Gradient Algorithms

no code implementations3 Oct 2024 Bin Gu, Xiyuan Wei, Hualin Zhang, Yi Chang, Heng Huang

While the random ZO estimator introduces bigger error and makes convergence analysis more challenging compared to coordinated ZO estimator, it requires only $\mathcal{O}(1)$ computation, which is significantly less than $\mathcal{O}(d)$ computation of the coordinated ZO estimator, with $d$ being dimension of the problem space.

Mixture of Efficient Diffusion Experts Through Automatic Interval and Sub-Network Selection

no code implementations23 Sep 2024 Alireza Ganjdanesh, Yan Kang, Yuchen Liu, Richard Zhang, Zhe Lin, Heng Huang

Finally, with a selected configuration, we fine-tune our pruned experts to obtain our mixture of efficient experts.

Denoising

From Lists to Emojis: How Format Bias Affects Model Alignment

no code implementations18 Sep 2024 Xuanchang Zhang, Wei Xiong, Lichang Chen, Tianyi Zhou, Heng Huang, Tong Zhang

In this work, we extend the study of biases in preference learning beyond the commonly recognized length bias, offering a comprehensive analysis of a wider range of format biases.

Chatbot

MLPHand: Real Time Multi-View 3D Hand Mesh Reconstruction via MLP Modeling

no code implementations23 Jun 2024 Jian Yang, Jiakun Li, Guoming Li, Zhen Shen, Huai-Yu Wu, Zhaoxin Fan, Heng Huang

Multi-view hand mesh reconstruction is a critical task for applications in virtual reality and human-computer interaction, but it remains a formidable challenge.

Brain Dialogue Interface (BDI): A User-Friendly fMRI Model for Interactive Brain Decoding

no code implementations17 Jun 2024 Heng Huang, Lin Zhao, Zihao Wu, Xiaowei Yu, Jing Zhang, Xintao Hu, Dajiang Zhu, Tianming Liu

To address this issue, this study introduces a user-friendly decoding model that enables dynamic communication with the brain, as opposed to the static decoding approaches utilized by traditional studies.

Brain Decoding

Not All Prompts Are Made Equal: Prompt-based Pruning of Text-to-Image Diffusion Models

1 code implementation17 Jun 2024 Alireza Ganjdanesh, Reza Shirkavand, Shangqian Gao, Heng Huang

Each architecture code represents a specialized model tailored to the prompts assigned to it, and the number of codes is a hyperparameter.

Contrastive Learning Image Generation

OPTune: Efficient Online Preference Tuning

no code implementations11 Jun 2024 Lichang Chen, Jiuhai Chen, Chenxi Liu, John Kirchenbauer, Davit Soselia, Chen Zhu, Tom Goldstein, Tianyi Zhou, Heng Huang

In this paper, we propose a more efficient data exploration strategy for online preference tuning (OPTune), which does not rely on human-curated or pre-collected teacher responses but dynamically samples informative responses for on-policy preference alignment.

Instruction Following

Event3DGS: Event-Based 3D Gaussian Splatting for High-Speed Robot Egomotion

no code implementations5 Jun 2024 Tianyi Xiong, Jiayi Wu, Botao He, Cornelia Fermuller, Yiannis Aloimonos, Heng Huang, Christopher A. Metzler

By combining differentiable rendering with explicit point-based scene representations, 3D Gaussian Splatting (3DGS) has demonstrated breakthrough 3D reconstruction capabilities.

3D Reconstruction 3D Scene Reconstruction +1

Interpretable Spatio-Temporal Embedding for Brain Structural-Effective Network with Ordinary Differential Equation

no code implementations21 May 2024 Haoteng Tang, Guodong Liu, Siyuan Dai, Kai Ye, Kun Zhao, Wenlu Wang, Carl Yang, Lifang He, Alex Leow, Paul Thompson, Heng Huang, Liang Zhan

The MRI-derived brain network serves as a pivotal instrument in elucidating both the structural and functional aspects of the brain, encompassing the ramifications of diseases and developmental processes.

Graph Learning

Jointly Training and Pruning CNNs via Learnable Agent Guidance and Alignment

no code implementations CVPR 2024 Alireza Ganjdanesh, Shangqian Gao, Heng Huang

We address this challenge by designing a mechanism to model the complex changing dynamics of the reward function and provide a representation of it to the RL agent.

Reinforcement Learning (RL)

Auto-Train-Once: Controller Network Guided Automatic Network Pruning from Scratch

1 code implementation CVPR 2024 Xidong Wu, Shangqian Gao, Zeyu Zhang, Zhenzhen Li, Runxue Bao, yanfu Zhang, Xiaoqian Wang, Heng Huang

Current techniques for deep neural network (DNN) pruning often involve intricate multi-step processes that require domain-specific expertise, making their widespread adoption challenging.

Network Pruning

A Unified and General Framework for Continual Learning

1 code implementation20 Mar 2024 Zhenyi Wang, Yan Li, Li Shen, Heng Huang

Extensive experiments on CL benchmarks and theoretical analysis demonstrate the effectiveness of the proposed refresh learning.

Continual Learning

Few-Shot Class Incremental Learning with Attention-Aware Self-Adaptive Prompt

1 code implementation14 Mar 2024 Chenxi Liu, Zhenyi Wang, Tianyi Xiong, Ruibo Chen, Yihan Wu, Junfeng Guo, Heng Huang

Few-Shot Class-Incremental Learning (FSCIL) models aim to incrementally learn new classes with scarce samples while preserving knowledge of old ones.

class-incremental learning Few-Shot Class-Incremental Learning +1

ODIN: Disentangled Reward Mitigates Hacking in RLHF

no code implementations11 Feb 2024 Lichang Chen, Chen Zhu, Davit Soselia, Jiuhai Chen, Tianyi Zhou, Tom Goldstein, Heng Huang, Mohammad Shoeybi, Bryan Catanzaro

In this work, we study the issue of reward hacking on the response length, a challenge emerging in Reinforcement Learning from Human Feedback (RLHF) on LLMs.

Device-Wise Federated Network Pruning

1 code implementation CVPR 2024 Shangqian Gao, Junyi Li, Zeyu Zhang, yanfu Zhang, Weidong Cai, Heng Huang

Neural network pruning particularly channel pruning is a widely used technique for compressing deep learning models to enable their deployment on edge devices with limited resources.

Edge-computing Federated Learning +1

BilevelPruning: Unified Dynamic and Static Channel Pruning for Convolutional Neural Networks

no code implementations CVPR 2024 Shangqian Gao, yanfu Zhang, Feihu Huang, Heng Huang

Most existing dynamic or runtime channel pruning methods have to store all weights to achieve efficient inference which brings extra storage costs.

Federated Continual Novel Class Learning

no code implementations21 Dec 2023 Lixu Wang, Chenxi Liu, Junfeng Guo, Jiahua Dong, Xiao Wang, Heng Huang, Qi Zhu

In a privacy-focused era, Federated Learning (FL) has emerged as a promising machine learning technique.

Federated Learning Novel Class Discovery +1

On the Role of Server Momentum in Federated Learning

no code implementations19 Dec 2023 Jianhui Sun, Xidong Wu, Heng Huang, Aidong Zhang

To our best knowledge, this is the first work that thoroughly analyzes the performances of server momentum with a hyperparameter scheduler and system heterogeneity.

Federated Learning

Predicting Scores of Various Aesthetic Attribute Sets by Learning from Overall Score Labels

no code implementations6 Dec 2023 Heng Huang, Xin Jin, Yaqi Liu, Hao Lou, Chaoen Xiao, Shuai Cui, Xinning Li, Dongqing Zou

Then, we define an aesthetic attribute contribution to describe the role of aesthetic attributes throughout an image and use it with the attribute scores and the overall scores to train our F2S model.

Attribute

Newton-CG methods for nonconvex unconstrained optimization with Hölder continuous Hessian

no code implementations22 Nov 2023 Chuan He, Heng Huang, Zhaosong Lu

In this paper we consider a nonconvex unconstrained optimization problem minimizing a twice differentiable objective function with H\"older continuous Hessian.

Token-Level Adversarial Prompt Detection Based on Perplexity Measures and Contextual Information

no code implementations20 Nov 2023 Zhengmian Hu, Gang Wu, Saayan Mitra, Ruiyi Zhang, Tong Sun, Heng Huang, Viswanathan Swaminathan

Our work aims to address this concern by introducing a novel approach to detecting adversarial prompts at a token level, leveraging the LLM's capability to predict the next token's probability.

GPT-4 Vision on Medical Image Classification -- A Case Study on COVID-19 Dataset

no code implementations27 Oct 2023 Ruibo Chen, Tianyi Xiong, Yihan Wu, Guodong Liu, Zhengmian Hu, Lichang Chen, Yanshuo Chen, Chenxi Liu, Heng Huang

This technical report delves into the application of GPT-4 Vision (GPT-4V) in the nuanced realm of COVID-19 image classification, leveraging the transformative potential of in-context learning to enhance diagnostic processes.

Image Classification In-Context Learning +1

Reflection-Tuning: Data Recycling Improves LLM Instruction-Tuning

2 code implementations18 Oct 2023 Ming Li, Lichang Chen, Jiuhai Chen, Shwai He, Heng Huang, Jiuxiang Gu, Tianyi Zhou

Recent advancements in Large Language Models (LLMs) have expanded the horizons of natural language understanding and generation.

Natural Language Understanding

A Resilient and Accessible Distribution-Preserving Watermark for Large Language Models

2 code implementations11 Oct 2023 Yihan Wu, Zhengmian Hu, Junfeng Guo, Hongyang Zhang, Heng Huang

Watermarking techniques offer a promising way to identify machine-generated content via embedding covert information into the contents generated from language models.

Language Modeling Language Modelling

Solving a Class of Non-Convex Minimax Optimization in Federated Learning

1 code implementation NeurIPS 2023 Xidong Wu, Jianhui Sun, Zhengmian Hu, Aidong Zhang, Heng Huang

We propose FL algorithms (FedSGDA+ and FedSGDA-M) and reduce existing complexity results for the most common minimax problems.

Federated Learning

Shielding the Unseen: Privacy Protection through Poisoning NeRF with Spatial Deformation

no code implementations4 Oct 2023 Yihan Wu, Brandon Y. Feng, Heng Huang

In this paper, we introduce an innovative method of safeguarding user privacy against the generative capabilities of Neural Radiance Fields (NeRF) models.

3D Scene Reconstruction Privacy Preserving

Towards Green AI in Fine-tuning Large Language Models via Adaptive Backpropagation

1 code implementation22 Sep 2023 Kai Huang, Hanyun Yin, Heng Huang, Wei Gao

With the fast growth of LLM-enabled AI applications and democratization of open-souced LLMs, fine-tuning has become possible for non-expert individuals, but intensively performed LLM fine-tuning worldwide could result in significantly high energy consumption and carbon footprint, which may bring large environmental impact.

Abstractive Text Summarization

Deep Prompt Tuning for Graph Transformers

no code implementations18 Sep 2023 Reza Shirkavand, Heng Huang

We propose a novel approach called deep graph prompt tuning as an alternative to fine-tuning for leveraging large graph transformer models in downstream graph based prediction tasks.

Graph Representation Learning

Serverless Federated AUPRC Optimization for Multi-Party Collaborative Imbalanced Data Mining

1 code implementation6 Aug 2023 Xidong Wu, Zhengmian Hu, Jian Pei, Heng Huang

To address the above challenge, we study the serverless multi-party collaborative AUPRC maximization problem since serverless multi-party collaborative training can cut down the communications cost by avoiding the server node bottleneck, and reformulate it as a conditional stochastic optimization problem in a serverless multi-party collaborative learning setting and propose a new ServerLess biAsed sTochastic gradiEnt (SLATE) algorithm to directly optimize the AUPRC.

Federated Learning Stochastic Optimization

AlpaGasus: Training A Better Alpaca with Fewer Data

3 code implementations17 Jul 2023 Lichang Chen, Shiyang Li, Jun Yan, Hai Wang, Kalpa Gunaratna, Vikas Yadav, Zheng Tang, Vijay Srinivasan, Tianyi Zhou, Heng Huang, Hongxia Jin

Large language models (LLMs) strengthen instruction-following capability through instruction-finetuning (IFT) on supervised instruction/response data.

Instruction Following

A Comprehensive Survey of Forgetting in Deep Learning Beyond Continual Learning

1 code implementation16 Jul 2023 Zhenyi Wang, Enneng Yang, Li Shen, Heng Huang

Through this comprehensive survey, we aspire to uncover potential solutions by drawing upon ideas and approaches from various fields that have dealt with forgetting.

Continual Learning Federated Learning +2

Cooperation or Competition: Avoiding Player Domination for Multi-Target Robustness via Adaptive Budgets

no code implementations CVPR 2023 Yimu Wang, Dinghuai Zhang, Yihan Wu, Heng Huang, Hongyang Zhang

We identify a phenomenon named player domination in the bargaining game, namely that the existing max-based approaches, such as MAX and MSD, do not converge.

InstructZero: Efficient Instruction Optimization for Black-Box Large Language Models

2 code implementations5 Jun 2023 Lichang Chen, Jiuhai Chen, Tom Goldstein, Heng Huang, Tianyi Zhou

Large language models~(LLMs) are instruction followers, but it can be challenging to find the best instruction for different situations, especially for black-box LLMs on which backpropagation is forbidden.

Bayesian Optimization

Prediction of Post-Operative Renal and Pulmonary Complications Using Transformers

no code implementations1 Jun 2023 Reza Shirkavand, Fei Zhang, Heng Huang

This work highlights the potential of deep learning techniques, specifically transformer-based models, in revolutionizing the healthcare industry's approach to postoperative care.

Deep Learning Management

Incomplete Multimodal Learning for Complex Brain Disorders Prediction

no code implementations25 May 2023 Reza Shirkavand, Liang Zhan, Heng Huang, Li Shen, Paul M. Thompson

Especially in studies of brain diseases, research cohorts may include both neuroimaging data and genetic data, but for practical clinical diagnosis, we often need to make disease predictions only based on neuroimages.

Data Integration

Prompting Language-Informed Distribution for Compositional Zero-Shot Learning

1 code implementation23 May 2023 Wentao Bao, Lichang Chen, Heng Huang, Yu Kong

Compositional zero-shot learning (CZSL) task aims to recognize unseen compositional visual concepts, e. g., sliced tomatoes, where the model is learned only from the seen compositions, e. g., sliced potatoes and red tomatoes.

Compositional Zero-Shot Learning Informativeness +1

PTP: Boosting Stability and Performance of Prompt Tuning with Perturbation-Based Regularizer

no code implementations3 May 2023 Lichang Chen, Heng Huang, Minhao Cheng

To address this critical problem, we first investigate and find that the loss landscape of vanilla prompt tuning is precipitous when it is visualized, where a slight change of input data can cause a big fluctuation in the loss landscape.

Natural Language Understanding

Backdoor Learning on Sequence to Sequence Models

no code implementations3 May 2023 Lichang Chen, Minhao Cheng, Heng Huang

Backdoor learning has become an emerging research area towards building a trustworthy machine learning system.

Machine Translation Sentence +3

When do you need Chain-of-Thought Prompting for ChatGPT?

no code implementations6 Apr 2023 Jiuhai Chen, Lichang Chen, Heng Huang, Tianyi Zhou

However, it is not clear whether CoT is still effective on more recent instruction finetuned (IFT) LLMs such as ChatGPT.

Arithmetic Reasoning Memorization

Communication-Efficient Federated Bilevel Optimization with Local and Global Lower Level Problems

no code implementations13 Feb 2023 Junyi Li, Feihu Huang, Heng Huang

In this work, we investigate Federated Bilevel Optimization problems and propose a communication-efficient algorithm, named FedBiOAcc.

Bilevel Optimization Federated Learning +1

FedDA: Faster Framework of Local Adaptive Gradient Methods via Restarted Dual Averaging

no code implementations13 Feb 2023 Junyi Li, Feihu Huang, Heng Huang

This matches the best known rate for first-order FL algorithms and \textbf{FedDA-MVR} is the first adaptive FL algorithm that achieves this rate.

Federated Learning

Decentralized Riemannian Algorithm for Nonconvex Minimax Problems

no code implementations8 Feb 2023 Xidong Wu, Zhengmian Hu, Heng Huang

The minimax optimization over Riemannian manifolds (possibly nonconvex constraints) has been actively applied to solve many problems, such as robust dimensionality reduction and deep neural networks with orthogonal weights (Stiefel manifold).

Dimensionality Reduction

A Newton-CG based barrier-augmented Lagrangian method for general nonconvex conic optimization

no code implementations10 Jan 2023 Chuan He, Heng Huang, Zhaosong Lu

In this paper we consider finding an approximate second-order stationary point (SOSP) of general nonconvex conic optimization that minimizes a twice differentiable function subject to nonlinear equality constraints and also a convex conic constraint.

Learning with Diversity: Self-Expanded Equalization for Better Generalized Deep Metric Learning

no code implementations ICCV 2023 Jiexi Yan, Zhihui Yin, Erkun Yang, Yanhua Yang, Heng Huang

Most existing DML methods focus on improving the model robustness against category shift to keep the performance on unseen categories.

Diversity Metric Learning

Structural Alignment for Network Pruning through Partial Regularization

no code implementations ICCV 2023 Shangqian Gao, Zeyu Zhang, yanfu Zhang, Feihu Huang, Heng Huang

To mitigate this gap, we first learn a target sub-network during the model training process, and then we use this sub-network to guide the learning of model weights through partial regularization.

Network Pruning

Faster Adaptive Federated Learning

no code implementations2 Dec 2022 Xidong Wu, Feihu Huang, Zhengmian Hu, Heng Huang

Federated learning has attracted increasing attention with the emergence of distributed data.

Federated Learning Image Classification +2

Towards Robust Dataset Learning

1 code implementation19 Nov 2022 Yihan Wu, Xinda Li, Florian Kerschbaum, Heng Huang, Hongyang Zhang

In this paper, we study the problem of learning a robust dataset such that any classifier naturally trained on the dataset is adversarially robust.

BI AVAN: Brain inspired Adversarial Visual Attention Network

no code implementations27 Oct 2022 Heng Huang, Lin Zhao, Xintao Hu, Haixing Dai, Lu Zhang, Dajiang Zhu, Tianming Liu

Visual attention is a fundamental mechanism in the human brain, and it inspires the design of attention mechanisms in deep neural networks.

FedGRec: Federated Graph Recommender System with Lazy Update of Latent Embeddings

no code implementations25 Oct 2022 Junyi Li, Heng Huang

Therefore, Federated Recommender (FedRec) systems are proposed to mitigate privacy concerns to non-distributed recommender systems.

Federated Learning Recommendation Systems

Communication-Efficient Adam-Type Algorithms for Distributed Data Mining

no code implementations14 Oct 2022 Wenhan Xian, Feihu Huang, Heng Huang

In our theoretical analysis, we prove that our new algorithm achieves a fast convergence rate of $O(\frac{1}{\sqrt{nT}} + \frac{1}{(k/d)^2 T})$ with the communication cost of $O(k \log(d))$ at each iteration.

Vocal Bursts Type Prediction

Interpretations Steered Network Pruning via Amortized Inferred Saliency Maps

1 code implementation7 Sep 2022 Alireza Ganjdanesh, Shangqian Gao, Heng Huang

To fill in this gap, we propose to address the channel pruning problem from a novel perspective by leveraging the interpretations of a model to steer the pruning process, thereby utilizing information from both inputs and outputs of the model.

Inductive Bias Network Pruning

An Accelerated Doubly Stochastic Gradient Method with Faster Explicit Model Identification

no code implementations11 Aug 2022 Runxue Bao, Bin Gu, Heng Huang

To address this challenge, we propose a novel accelerated doubly stochastic gradient descent (ADSGD) method for sparsity regularized loss minimization problems, which can reduce the number of block iterations by eliminating inactive coefficients during the optimization process and eventually achieve faster explicit model identification and improve the algorithm efficiency.

Dimensionality Reduction

Aesthetic Language Guidance Generation of Images Using Attribute Comparison

no code implementations9 Aug 2022 Xin Jin, Qiang Deng, Jianwen Lv, Heng Huang, Hao Lou, Chaoen Xiao

The differences of the three attributes between the input images and the photography templates or the guidance images are described in natural language, which is aesthetic natural language guidance (ALG).

Attribute

Aesthetic Attributes Assessment of Images with AMANv2 and DPC-CaptionsV2

no code implementations9 Aug 2022 Xinghui Zhou, Xin Jin, Jianwen Lv, Heng Huang, Ming Mao, Shuai Cui

In this paper, we propose aesthetic attribute assessment, which is the aesthetic attributes captioning, i. e., to assess the aesthetic attributes such as composition, lighting usage and color arrangement.

Attribute Image Captioning

Contrastive Brain Network Learning via Hierarchical Signed Graph Pooling Model

no code implementations14 Jul 2022 Haoteng Tang, Guixiang Ma, Lei Guo, Xiyao Fu, Heng Huang, Liang Zhang

Here, we propose an interpretable hierarchical signed graph representation learning model to extract graph-level representations from brain functional networks, which can be used for different prediction tasks.

Contrastive Learning Graph Learning +1

Balanced Self-Paced Learning for AUC Maximization

no code implementations8 Jul 2022 Bin Gu, Chenkang Zhang, Huan Xiong, Heng Huang

Self-paced learning is an effective method for handling noisy data.

RetrievalGuard: Provably Robust 1-Nearest Neighbor Image Retrieval

no code implementations17 Jun 2022 Yihan Wu, Hongyang Zhang, Heng Huang

The challenge is to design a provably robust algorithm that takes into consideration the 1-NN search and the high-dimensional nature of the embedding space.

Image Retrieval Retrieval

Communication-Efficient Robust Federated Learning with Noisy Labels

no code implementations11 Jun 2022 Junyi Li, Jian Pei, Heng Huang

Bilevel optimization problem is a type of optimization problem with two levels of entangled problems.

Bilevel Optimization Federated Learning +2

Functional2Structural: Cross-Modality Brain Networks Representation Learning

no code implementations6 May 2022 Haoteng Tang, Xiyao Fu, Lei Guo, Yalin Wang, Scott Mackin, Olusola Ajilore, Alex Leow, Paul Thompson, Heng Huang, Liang Zhan

Since brain networks derived from functional and structural MRI describe the brain topology from different perspectives, exploring a representation that combines these cross-modality brain networks is non-trivial.

Disease Prediction Graph Learning +2

Local Stochastic Bilevel Optimization with Momentum-Based Variance Reduction

no code implementations3 May 2022 Junyi Li, Feihu Huang, Heng Huang

Specifically, we first propose the FedBiO, a deterministic gradient-based algorithm and we show it requires $O(\epsilon^{-2})$ number of iterations to reach an $\epsilon$-stationary point.

BIG-bench Machine Learning Bilevel Optimization +3

Distributed Dynamic Safe Screening Algorithms for Sparse Regularization

no code implementations23 Apr 2022 Runxue Bao, Xidong Wu, Wenhan Xian, Heng Huang

To the best of our knowledge, this is the first work of distributed safe dynamic screening method.

Distributed Optimization

Desirable Companion for Vertical Federated Learning: New Zeroth-Order Gradient Based Algorithm

no code implementations19 Mar 2022 Qingsong Zhang, Bin Gu, Zhiyuan Dang, Cheng Deng, Heng Huang

Based on that, we propose a novel and practical VFL framework with black-box models, which is inseparably interconnected to the promising properties of ZOO.

Vertical Federated Learning

Closing the Generalization Gap of Cross-silo Federated Medical Image Segmentation

no code implementations CVPR 2022 An Xu, Wenqi Li, Pengfei Guo, Dong Yang, Holger Roth, Ali Hatamizadeh, Can Zhao, Daguang Xu, Heng Huang, Ziyue Xu

In this work, we propose a novel training framework FedSM to avoid the client drift issue and successfully close the generalization gap compared with the centralized training for medical image segmentation tasks for the first time.

Federated Learning Image Segmentation +3

Towards Bi-directional Skip Connections in Encoder-Decoder Architectures and Beyond

no code implementations11 Mar 2022 Tiange Xiang, Chaoyi Zhang, Xinyi Wang, Yang song, Dongnan Liu, Heng Huang, Weidong Cai

With the backward skip connections, we propose a U-Net based network family, namely Bi-directional O-shape networks, which set new benchmarks on multiple public medical imaging segmentation datasets.

Decoder Medical Image Analysis +3

A Law of Robustness beyond Isoperimetry

no code implementations23 Feb 2022 Yihan Wu, Heng Huang, Hongyang Zhang

We prove a Lipschitzness lower bound $\Omega(\sqrt{n/p})$ of the interpolating neural network with $p$ parameters on arbitrary data distributions.

HNF-Netv2 for Brain Tumor Segmentation using multi-modal MR Imaging

no code implementations10 Feb 2022 Haozhe Jia, Chao Bai, Weidong Cai, Heng Huang, Yong Xia

In our previous work, $i. e.$, HNF-Net, high-resolution feature representation and light-weight non-local self-attention mechanism are exploited for brain tumor segmentation using multi-modal MR imaging.

Brain Tumor Segmentation Segmentation +1

Decompose to Adapt: Cross-domain Object Detection via Feature Disentanglement

1 code implementation6 Jan 2022 Dongnan Liu, Chaoyi Zhang, Yang song, Heng Huang, Chenyu Wang, Michael Barnett, Weidong Cai

Recent advances in unsupervised domain adaptation (UDA) techniques have witnessed great success in cross-domain computer vision tasks, enhancing the generalization ability of data-driven deep learning architectures by bridging the domain distribution gaps.

Disentanglement object-detection +3

Noise Is Also Useful: Negative Correlation-Steered Latent Contrastive Learning

no code implementations CVPR 2022 Jiexi Yan, Lei Luo, Chenghao Xu, Cheng Deng, Heng Huang

While in metric space, we utilize weakly-supervised contrastive learning to excavate these negative correlations hidden in noisy data.

Contrastive Learning Diversity

A Fully Single Loop Algorithm for Bilevel Optimization without Hessian Inverse

no code implementations9 Dec 2021 Junyi Li, Bin Gu, Heng Huang

Combining our new formulation with the alternative update of the inner and outer variables, we propose an efficient fully single loop algorithm.

Bilevel Optimization

Fast Training Method for Stochastic Compositional Optimization Problems

no code implementations NeurIPS 2021 Hongchang Gao, Heng Huang

The stochastic compositional optimization problem covers a wide range of machine learning models, such as sparse additive models and model-agnostic meta-learning.

Additive models Meta-Learning

Optimal Underdamped Langevin MCMC Method

no code implementations NeurIPS 2021 Zhengmian Hu, Feihu Huang, Heng Huang

In the paper, we study the underdamped Langevin diffusion (ULD) with strongly-convex potential consisting of finite summation of $N$ smooth components, and propose an efficient discretization method, which requires $O(N+d^\frac{1}{3}N^\frac{2}{3}/\varepsilon^\frac{2}{3})$ gradient evaluations to achieve $\varepsilon$-error (in $\sqrt{\mathbb{E}{\lVert{\cdot}\rVert_2^2}}$ distance) for approximating $d$-dimensional ULD.

Efficient Mirror Descent Ascent Methods for Nonsmooth Minimax Problems

no code implementations NeurIPS 2021 Feihu Huang, Xidong Wu, Heng Huang

For our stochastic algorithms, we first prove that the mini-batch stochastic mirror descent ascent (SMDA) method obtains a sample complexity of $O(\kappa^3\epsilon^{-4})$ for finding an $\epsilon$-stationary point, where $\kappa$ denotes the condition number.

A Faster Decentralized Algorithm for Nonconvex Minimax Problems

no code implementations NeurIPS 2021 Wenhan Xian, Feihu Huang, yanfu Zhang, Heng Huang

We prove that our DM-HSGD algorithm achieves stochastic first-order oracle (SFO) complexity of $O(\kappa^3 \epsilon^{-3})$ for decentralized stochastic nonconvex-strongly-concave problem to search an $\epsilon$-stationary point, which improves the exiting best theoretical results.

Adaptive Hierarchical Similarity Metric Learning with Noisy Labels

no code implementations29 Oct 2021 Jiexi Yan, Lei Luo, Cheng Deng, Heng Huang

Since these noisy labels often cause severe performance degradation, it is crucial to enhance the robustness and generalization ability of DML.

Learning with noisy labels Metric Learning

Efficient Semi-Supervised Adversarial Training without Guessing Labels

no code implementations29 Sep 2021 Huimin Wu, Heng Huang, Bin Gu

To adapt to semi-supervised learning problems, they need to estimate labels for unlabeled data in advance, which inevitably degenerates the performance of the learned model due to the bias on the estimation of labels for unlabeled data.

Efficient Bi-level Optimization for Non-smooth Optimization

no code implementations29 Sep 2021 Wanli Shi, Heng Huang, Bin Gu

Then, we transform the smoothed bi-level optimization to an unconstrained penalty problem by replacing the smoothed sub-problem with its first-order necessary conditions.

Open-Ended Question Answering

Understanding Metric Learning on Unit Hypersphere and Generating Better Examples for Adversarial Training

no code implementations29 Sep 2021 Yihan Wu, Heng Huang

In this paper, we boost the performance of deep metric learning (DML) models with adversarial examples generated by attacking two new objective functions: \textit{intra-class alignment} and \textit{hyperspherical uniformity}.

Metric Learning Representation Learning +1

Adversarial Fairness Network

no code implementations29 Sep 2021 Taeuk Jang, Xiaoqian Wang, Heng Huang

To achieve this goal, we reformulate the data input by eliminating the sensitive information and strengthen model fairness by minimizing the marginal contribution of the sensitive feature.

BIG-bench Machine Learning Fairness

AsySQN: Faster Vertical Federated Learning Algorithms with Better Computation Resource Utilization

no code implementations26 Sep 2021 Qingsong Zhang, Bin Gu, Cheng Deng, Songxiang Gu, Liefeng Bo, Jian Pei, Heng Huang

To address the challenges of communication and computation resource utilization, we propose an asynchronous stochastic quasi-Newton (AsySQN) framework for VFL, under which three algorithms, i. e. AsySQN-SGD, -SVRG and -SAGA, are proposed.

Privacy Preserving Vertical Federated Learning

An Accelerated Variance-Reduced Conditional Gradient Sliding Algorithm for First-order and Zeroth-order Optimization

no code implementations18 Sep 2021 Xiyuan Wei, Bin Gu, Heng Huang

The conditional gradient algorithm (also known as the Frank-Wolfe algorithm) has recently regained popularity in the machine learning community due to its projection-free property to solve constrained problems.

DSNet: A Dual-Stream Framework for Weakly-Supervised Gigapixel Pathology Image Analysis

no code implementations13 Sep 2021 Tiange Xiang, Yang song, Chaoyi Zhang, Dongnan Liu, Mei Chen, Fan Zhang, Heng Huang, Lauren O'Donnell, Weidong Cai

With image-level labels only, patch-wise classification would be sub-optimal due to inconsistency between the patch appearance and image-level label.

Classification whole slide images

Boundary-aware Graph Reasoning for Semantic Segmentation

no code implementations9 Aug 2021 Haoteng Tang, Haozhe Jia, Weidong Cai, Heng Huang, Yong Xia, Liang Zhan

In this paper, we propose a Boundary-aware Graph Reasoning (BGR) module to learn long-range contextual features for semantic segmentation.

graph construction Segmentation +1

PSGR: Pixel-wise Sparse Graph Reasoning for COVID-19 Pneumonia Segmentation in CT Images

no code implementations9 Aug 2021 Haozhe Jia, Haoteng Tang, Guixiang Ma, Weidong Cai, Heng Huang, Liang Zhan, Yong Xia

In the PSGR module, a graph is first constructed by projecting each pixel on a node based on the features produced by the segmentation backbone, and then converted into a sparsely-connected graph by keeping only K strongest connections to each uncertain pixel.

Computed Tomography (CT) graph construction +3

Enhanced Bilevel Optimization via Bregman Distance

no code implementations26 Jul 2021 Feihu Huang, Junyi Li, Shangqian Gao, Heng Huang

Specifically, we propose a bilevel optimization method based on Bregman distance (BiO-BreD) to solve deterministic bilevel problems, which achieves a lower computational complexity than the best known results.

Bilevel Optimization Hyperparameter Optimization +2

BiX-NAS: Searching Efficient Bi-directional Architecture for Medical Image Segmentation

1 code implementation26 Jun 2021 Xinyi Wang, Tiange Xiang, Chaoyi Zhang, Yang song, Dongnan Liu, Heng Huang, Weidong Cai

We evaluate BiX-NAS on two segmentation tasks using three different medical image datasets, and the experimental results show that our BiX-NAS searched architecture achieves the state-of-the-art performance with significantly lower computational cost.

Image Segmentation Medical Image Segmentation +3

Bregman Gradient Policy Optimization

1 code implementation ICLR 2022 Feihu Huang, Shangqian Gao, Heng Huang

In the paper, we design a novel Bregman gradient policy optimization framework for reinforcement learning based on Bregman divergences and momentum techniques.

reinforcement-learning Reinforcement Learning +1

Network Pruning via Performance Maximization

1 code implementation CVPR 2021 Shangqian Gao, Feihu Huang, Weidong Cai, Heng Huang

Specifically, we train a stand-alone neural network to predict sub-networks' performance and then maximize the output of the network as a proxy of accuracy to guide pruning.

Model Compression Network Pruning

Nearest Neighbor Matching for Deep Clustering

1 code implementation CVPR 2021 Zhiyuan Dang, Cheng Deng, Xu Yang, Kun Wei, Heng Huang

Specifically, for the local level, we match the nearest neighbors based on batch embedded features, as for the global one, we match neighbors from overall embedded features.

Clustering Deep Clustering

Unsupervised Hyperbolic Metric Learning

no code implementations CVPR 2021 Jiexi Yan, Lei Luo, Cheng Deng, Heng Huang

Learning feature embedding directly from images without any human supervision is a very challenging and essential task in the field of computer vision and machine learning.

Anatomy Metric Learning

SUPER-ADAM: Faster and Universal Framework of Adaptive Gradients

1 code implementation NeurIPS 2021 Feihu Huang, Junyi Li, Heng Huang

To fill this gap, we propose a faster and universal framework of adaptive gradients (i. e., SUPER-ADAM) by introducing a universal adaptive matrix that includes most existing adaptive gradient forms.

Fast Training Method for Stochastic Compositional Optimization Problems

no code implementations NeurIPS 2021 Hongchang Gao, Heng Huang

The stochastic compositional optimization problem covers a wide range of machine learning models, such as sparse additive models and model-agnostic meta-learning.

Additive models Meta-Learning

Learning Sampling Policy for Faster Derivative Free Optimization

no code implementations9 Apr 2021 Zhou Zhai, Bin Gu, Heng Huang

To explore this problem, in this paper, we propose a new reinforcement learning based ZO algorithm (ZO-RL) with learning the sampling policy for generating the perturbations in ZO optimization instead of using random sampling.

reinforcement-learning Reinforcement Learning (RL)

Doubly Contrastive Deep Clustering

1 code implementation9 Mar 2021 Zhiyuan Dang, Cheng Deng, Xu Yang, Heng Huang

In this paper, we present a novel Doubly Contrastive Deep Clustering (DCDC) framework, which constructs contrastive loss over both sample and class views to obtain more discriminative features and competitive results.

Clustering Contrastive Learning +2

Data Augmentation for Object Detection via Differentiable Neural Rendering

1 code implementation4 Mar 2021 Guanghan Ning, Guang Chen, Chaowei Tan, Si Luo, Liefeng Bo, Heng Huang

We propose a new offline data augmentation method for object detection, which semantically interpolates the training data with novel views.

Data Augmentation Neural Rendering +4

Secure Bilevel Asynchronous Vertical Federated Learning with Backward Updating

no code implementations1 Mar 2021 Qingsong Zhang, Bin Gu, Cheng Deng, Heng Huang

Vertical federated learning (VFL) attracts increasing attention due to the emerging demands of multi-party collaborative modeling and concerns of privacy leakage.

Vertical Federated Learning

Optimizing Large-Scale Hyperparameters via Automated Learning Algorithm

1 code implementation17 Feb 2021 Bin Gu, Guodong Liu, yanfu Zhang, Xiang Geng, Heng Huang

Modern machine learning algorithms usually involve tuning multiple (from one to thousands) hyperparameters which play a pivotal role in terms of model generalizability.

Hyperparameter Optimization

A New Framework for Variance-Reduced Hamiltonian Monte Carlo

no code implementations9 Feb 2021 Zhengmian Hu, Feihu Huang, Heng Huang

Moreover, our HMC methods with biased gradient estimators, such as SARAH and SARGE, require $\tilde{O}(N+\sqrt{N} \kappa^2 d^{\frac{1}{2}} \varepsilon^{-1})$ gradient complexity, which has the same dependency on condition number $\kappa$ and dimension $d$ as full gradient method, but improves the dependency of sample size $N$ for a factor of $N^\frac{1}{2}$.

Coordinating Momenta for Cross-silo Federated Learning

no code implementations8 Feb 2021 An Xu, Heng Huang

In this work, we propose a new method to improve the training performance in cross-silo FL via maintaining double momentum buffers.

Federated Learning

A Bayesian Federated Learning Framework with Online Laplace Approximation

no code implementations3 Feb 2021 Liangxi Liu, Xi Jiang, Feng Zheng, Hong Chen, Guo-Jun Qi, Heng Huang, Ling Shao

On the client side, a prior loss that uses the global posterior probabilistic parameters delivered from the server is designed to guide the local training.

Federated Learning

Model Compression via Hyper-Structure Network

no code implementations1 Jan 2021 Shangqian Gao, Feihu Huang, Heng Huang

In this paper, we propose a novel channel pruning method to solve the problem of compression and acceleration of Convolutional Neural Networks (CNNs).

Model Compression

Exploration and Estimation for Model Compression

no code implementations ICCV 2021 yanfu Zhang, Shangqian Gao, Heng Huang

In this paper, we focus on the discrimination-aware compression of Convolutional Neural Networks (CNNs).

Model Compression

Adversarial Attack on Deep Cross-Modal Hamming Retrieval

no code implementations ICCV 2021 Chao Li, Shangqian Gao, Cheng Deng, Wei Liu, Heng Huang

Specifically, given a target model, we first construct its substitute model to exploit cross-modal correlations within hamming space, with which we create adversarial examples by limitedly querying from a target model.

Adversarial Attack Cross-Modal Retrieval +2

Learning Better Visual Data Similarities via New Grouplet Non-Euclidean Embedding

no code implementations ICCV 2021 yanfu Zhang, Lei Luo, Wenhan Xian, Heng Huang

However, pair-wise methods involve expensive training costs, while proxy-based methods are less accurate in characterizing the relationships between data points.

Metric Learning

Delay-Tolerant Local SGD for Efficient Distributed Training

no code implementations1 Jan 2021 An Xu, Xiao Yan, Hongchang Gao, Heng Huang

The heavy communication for model synchronization is a major bottleneck for scaling up the distributed deep neural network training to many workers.

Federated Learning

CommPOOL: An Interpretable Graph Pooling Framework for Hierarchical Graph Representation Learning

no code implementations10 Dec 2020 Haoteng Tang, Guixiang Ma, Lifang He, Heng Huang, Liang Zhan

In this paper, we propose a new interpretable graph pooling framework - CommPOOL, that can capture and preserve the hierarchical community structure of graphs in the graph representation learning process.

Graph Classification Graph Representation Learning

A Deep Drift-Diffusion Model for Image Aesthetic Score Distribution Prediction

no code implementations15 Oct 2020 Xin Jin, Xiqiao Li, Heng Huang, XiaoDong Li, Xinghui Zhou

In this paper, we propose a Deep Drift-Diffusion (DDD) model inspired by psychologists to predict aesthetic score distribution from images.

Binary Classification

PDAM: A Panoptic-Level Feature Alignment Framework for Unsupervised Domain Adaptive Instance Segmentation in Microscopy Images

1 code implementation11 Sep 2020 Dongnan Liu, Donghao Zhang, Yang song, Fan Zhang, Lauren O'Donnell, Heng Huang, Mei Chen, Weidong Cai

In this work, we present an unsupervised domain adaptation (UDA) method, named Panoptic Domain Adaptive Mask R-CNN (PDAM), for unsupervised instance segmentation in microscopy images.

Instance Segmentation Segmentation +3

Improved Bilevel Model: Fast and Optimal Algorithm with Theoretical Guarantee

no code implementations1 Sep 2020 Junyi Li, Bin Gu, Heng Huang

In this paper, we propose an improved bilevel model which converges faster and better compared to the current formulation.

Representation Learning

Adaptive Serverless Learning

no code implementations24 Aug 2020 Hongchang Gao, Heng Huang

To the best of our knowledge, this is the first adaptive decentralized training approach.

Periodic Stochastic Gradient Descent with Momentum for Decentralized Training

no code implementations24 Aug 2020 Hongchang Gao, Heng Huang

The condition for achieving the linear speedup is also provided for this variant.

Accelerated Zeroth-Order and First-Order Momentum Methods from Mini to Minimax Optimization

no code implementations18 Aug 2020 Feihu Huang, Shangqian Gao, Jian Pei, Heng Huang

Our Acc-MDA achieves a low gradient complexity of $\tilde{O}(\kappa_y^{4. 5}\epsilon^{-3})$ without requiring large batches for finding an $\epsilon$-stationary point.

Adversarial Attack

Federated Doubly Stochastic Kernel Learning for Vertically Partitioned Data

no code implementations14 Aug 2020 Bin Gu, Zhiyuan Dang, Xiang Li, Heng Huang

In this paper, we focus on nonlinear learning with kernels, and propose a federated doubly stochastic kernel learning (FDSKL) algorithm for vertically partitioned data.

BIG-bench Machine Learning Federated Learning

Privacy-Preserving Asynchronous Federated Learning Algorithms for Multi-Party Vertically Collaborative Learning

no code implementations14 Aug 2020 Bin Gu, An Xu, Zhouyuan Huo, Cheng Deng, Heng Huang

To the best of our knowledge, AFSGD-VP and its SVRG and SAGA variants are the first asynchronous federated learning algorithms for vertically partitioned data.

Federated Learning Privacy Preserving

Step-Ahead Error Feedback for Distributed Training with Compressed Gradient

no code implementations13 Aug 2020 An Xu, Zhouyuan Huo, Heng Huang

Both our theoretical and empirical results show that our new methods can handle the "gradient mismatch" problem.

Faster Stochastic Alternating Direction Method of Multipliers for Nonconvex Optimization

no code implementations4 Aug 2020 Feihu Huang, Songcan Chen, Heng Huang

Our theoretical analysis shows that the online SPIDER-ADMM has the IFO complexity of $\mathcal{O}(\epsilon^{-\frac{3}{2}})$, which improves the existing best results by a factor of $\mathcal{O}(\epsilon^{-\frac{1}{2}})$.

Momentum-Based Policy Gradient Methods

1 code implementation ICML 2020 Feihu Huang, Shangqian Gao, Jian Pei, Heng Huang

In particular, we present a non-adaptive version of IS-MBPG method, i. e., IS-MBPG*, which also reaches the best known sample complexity of $O(\epsilon^{-3})$ without any large batches.

Policy Gradient Methods

BiO-Net: Learning Recurrent Bi-directional Connections for Encoder-Decoder Architecture

1 code implementation1 Jul 2020 Tiange Xiang, Chaoyi Zhang, Dongnan Liu, Yang song, Heng Huang, Weidong Cai

U-Net has become one of the state-of-the-art deep learning-based approaches for modern computer vision tasks such as semantic segmentation, super resolution, image denoising, and inpainting.

Decoder Image Denoising +3

Fast OSCAR and OWL Regression via Safe Screening Rules

1 code implementation29 Jun 2020 Runxue Bao, Bin Gu, Heng Huang

Moreover, we prove that the algorithms with our screening rule are guaranteed to have identical results with the original algorithms.

regression Sparse Learning

Faster Secure Data Mining via Distributed Homomorphic Encryption

no code implementations17 Jun 2020 Junyi Li, Heng Huang

Due to the rising privacy demand in data mining, Homomorphic Encryption (HE) is receiving more and more attention recently for its capability to do computations over the encrypted field.

Cloud Computing

Detached Error Feedback for Distributed SGD with Random Sparsification

no code implementations11 Apr 2020 An Xu, Heng Huang

To tackle this important issue, we improve the communication-efficient distributed SGD from a novel aspect, that is, the trade-off between the variance and second moment of the gradient.

Generalization Bounds Image Classification +1

Optimal Gradient Quantization Condition for Communication-Efficient Distributed Training

no code implementations25 Feb 2020 An Xu, Zhouyuan Huo, Heng Huang

The communication of gradients is costly for training deep neural networks with multiple devices in computer vision applications.

Quantization

Panoptic Feature Fusion Net: A Novel Instance Segmentation Paradigm for Biomedical and Biological Images

1 code implementation15 Feb 2020 Dongnan Liu, Donghao Zhang, Yang song, Heng Huang, Weidong Cai

Specifically, our proposed PFFNet contains a residual attention feature fusion mechanism to incorporate the instance prediction with the semantic features, in order to facilitate the semantic contextual information learning in the instance branch.

Instance Segmentation Medical Image Segmentation +2

Large Batch Training Does Not Need Warmup

1 code implementation4 Feb 2020 Zhouyuan Huo, Bin Gu, Heng Huang

Training deep neural networks using a large batch size has shown promising results and benefits many real-world applications.

Quadruply Stochastic Gradient Method for Large Scale Nonlinear Semi-Supervised Ordinal Regression AUC Optimization

no code implementations24 Dec 2019 Wanli Shi, Bin Gu, Xinag Li, Heng Huang

Semi-supervised ordinal regression (S$^2$OR) problems are ubiquitous in real-world applications, where only a few ordered instances are labeled and massive instances r