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.
no code implementations • ECCV 2020 • Jingwei Xin, Nannan Wang, Xinrui Jiang, Jie Li, Heng Huang, Xinbo Gao
Lighter model and faster inference are the focus of current single image super-resolution (SISR) research.
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.
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.
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.
no code implementations • 19 Dec 2024 • Reza Shirkavand, Peiran Yu, Shangqian Gao, Gowthami Somepalli, Tom Goldstein, Heng Huang
Recent advances in diffusion generative models have yielded remarkable progress.
1 code implementation • 19 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.
no code implementations • 6 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.
no code implementations • 22 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.
1 code implementation • 13 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.
no code implementations • 7 Nov 2024 • Junyi Li, Heng Huang
Bilevel Optimization has experienced significant advancements recently with the introduction of new efficient algorithms.
1 code implementation • 1 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.
no code implementations • 29 Oct 2024 • Zhengmian Hu, Tong Zheng, Heng Huang
Authorship attribution aims to identify the origin or author of a document.
no code implementations • 27 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.
no code implementations • 17 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.
no code implementations • 17 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).
no code implementations • 16 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).
no code implementations • 14 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.
no code implementations • 3 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.
no code implementations • 3 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.
no code implementations • 23 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.
no code implementations • 18 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.
1 code implementation • 14 Jul 2024 • Jianan Fan, Dongnan Liu, Canran Li, Hang Chang, Heng Huang, Filip Braet, Mei Chen, Weidong Cai
Cellular nuclei recognition serves as a fundamental and essential step in the workflow of digital pathology.
no code implementations • 23 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.
no code implementations • 17 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.
1 code implementation • 17 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.
no code implementations • 14 Jun 2024 • Vasu Singla, Kaiyu Yue, Sukriti Paul, Reza Shirkavand, Mayuka Jayawardhana, Alireza Ganjdanesh, Heng Huang, Abhinav Bhatele, Gowthami Somepalli, Tom Goldstein
Training large vision-language models requires extensive, high-quality image-text pairs.
no code implementations • 11 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.
no code implementations • 5 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.
no code implementations • 2 Jun 2024 • Yihan Wu, Ruibo Chen, Zhengmian Hu, Yanshuo Chen, Junfeng Guo, Hongyang Zhang, Heng Huang
Experimental results support that the beta-watermark can effectively reduce the distribution bias under key collisions.
no code implementations • 21 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.
no code implementations • 5 May 2024 • Xin Jin, Qianqian Qiao, Yi Lu, Shan Gao, Heng Huang, Guangdong Li
This dataset solely comprises overall scores for high-quality artistic images.
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.
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.
1 code implementation • 20 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.
1 code implementation • 14 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
1 code implementation • 5 Mar 2024 • Zhaoxin Fan, Runmin Jiang, Junhao Wu, Xin Huang, Tianyang Wang, Heng Huang, Min Xu
3D medical image segmentation is a challenging task with crucial implications for disease diagnosis and treatment planning.
no code implementations • CVPR 2024 • Jianan Fan, Dongnan Liu, Hang Chang, Heng Huang, Mei Chen, Weidong Cai
Machine learning holds tremendous promise for transforming the fundamental practice of scientific discovery by virtue of its data-driven nature.
1 code implementation • 19 Feb 2024 • Ruibo Chen, Yihan Wu, Lichang Chen, Guodong Liu, Qi He, Tianyi Xiong, Chenxi Liu, Junfeng Guo, Heng Huang
In the first stage, we devise a scoring network to evaluate the difficulty of training instructions, which is co-trained with the VLM.
no code implementations • 11 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.
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.
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.
no code implementations • 22 Dec 2023 • Alireza Ganjdanesh, Shangqian Gao, Hirad Alipanah, Heng Huang
Thus, they neglect the critical characteristic of GANs: their local density structure over their learned manifold.
no code implementations • 21 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.
no code implementations • 19 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.
no code implementations • 6 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.
no code implementations • 22 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.
no code implementations • 20 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.
no code implementations • 27 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.
2 code implementations • 18 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.
2 code implementations • 11 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.
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.
no code implementations • 4 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.
1 code implementation • 22 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.
no code implementations • 18 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.
no code implementations • 15 Sep 2023 • Marinka Zitnik, Michelle M. Li, Aydin Wells, Kimberly Glass, Deisy Morselli Gysi, Arjun Krishnan, T. M. Murali, Predrag Radivojac, Sushmita Roy, Anaïs Baudot, Serdar Bozdag, Danny Z. Chen, Lenore Cowen, Kapil Devkota, Anthony Gitter, Sara Gosline, Pengfei Gu, Pietro H. Guzzi, Heng Huang, Meng Jiang, Ziynet Nesibe Kesimoglu, Mehmet Koyuturk, Jian Ma, Alexander R. Pico, Nataša Pržulj, Teresa M. Przytycka, Benjamin J. Raphael, Anna Ritz, Roded Sharan, Yang shen, Mona Singh, Donna K. Slonim, Hanghang Tong, Xinan Holly Yang, Byung-Jun Yoon, Haiyuan Yu, Tijana Milenković
Network biology is an interdisciplinary field bridging computational and biological sciences that has proved pivotal in advancing the understanding of cellular functions and diseases across biological systems and scales.
1 code implementation • 6 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.
1 code implementation • ICCV 2023 • Jianan Fan, Dongnan Liu, Hang Chang, Heng Huang, Mei Chen, Weidong Cai
The success of automated medical image analysis depends on large-scale and expert-annotated training sets.
3 code implementations • 17 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.
1 code implementation • 16 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.
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.
2 code implementations • 5 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.
no code implementations • 1 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.
no code implementations • 25 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.
1 code implementation • 23 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.
no code implementations • 3 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.
no code implementations • 3 May 2023 • Lichang Chen, Minhao Cheng, Heng Huang
Backdoor learning has become an emerging research area towards building a trustworthy machine learning system.
no code implementations • 12 Apr 2023 • Xiangyu Xu, Lichang Chen, Changjiang Cai, Huangying Zhan, Qingan Yan, Pan Ji, Junsong Yuan, Heng Huang, Yi Xu
Direct optimization of interpolated features on multi-resolution voxel grids has emerged as a more efficient alternative to MLP-like modules.
no code implementations • 6 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.
no code implementations • 13 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.
no code implementations • 13 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.
no code implementations • 8 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).
no code implementations • 10 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.
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.
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.
1 code implementation • 9 Dec 2022 • Yihan Wu, Aleksandar Bojchevski, Heng Huang
In this paper, we extensively study this phenomenon for graph data.
no code implementations • 2 Dec 2022 • Xidong Wu, Feihu Huang, Zhengmian Hu, Heng Huang
Federated learning has attracted increasing attention with the emergence of distributed data.
1 code implementation • 19 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.
no code implementations • 27 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.
no code implementations • 25 Oct 2022 • Junyi Li, Heng Huang
Therefore, Federated Recommender (FedRec) systems are proposed to mitigate privacy concerns to non-distributed recommender systems.
no code implementations • 14 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.
1 code implementation • 7 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.
no code implementations • 11 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.
no code implementations • 9 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).
no code implementations • 9 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.
no code implementations • 14 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.
no code implementations • 8 Jul 2022 • Bin Gu, Chenkang Zhang, Huan Xiong, Heng Huang
Self-paced learning is an effective method for handling noisy data.
no code implementations • 17 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.
no code implementations • 11 Jun 2022 • Junyi Li, Jian Pei, Heng Huang
Bilevel optimization problem is a type of optimization problem with two levels of entangled problems.
no code implementations • 6 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.
no code implementations • 3 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.
no code implementations • 23 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.
no code implementations • 19 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.
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.
no code implementations • 11 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.
no code implementations • 23 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.
no code implementations • 10 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.
1 code implementation • 6 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.
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.
no code implementations • 9 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.
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.
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.
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.
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.
no code implementations • 29 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.
no code implementations • 29 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.
no code implementations • 29 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.
no code implementations • 29 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}.
no code implementations • 29 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.
no code implementations • 26 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.
no code implementations • 18 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.
no code implementations • 13 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.
no code implementations • 9 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.
no code implementations • 9 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.
no code implementations • 26 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.
3 code implementations • 8 Jul 2021 • Bo Liu, Chaowei Tan, Jiazhou Wang, Tao Zeng, Huasong Shan, Houpu Yao, Heng Huang, Peng Dai, Liefeng Bo, Yanqing Chen
We use this platform to demonstrate our research and development results on privacy preserving machine learning algorithms.
1 code implementation • 26 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.
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.
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.
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.
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.
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.
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.
no code implementations • 9 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.
1 code implementation • 9 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.
1 code implementation • 4 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.
no code implementations • 1 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.
1 code implementation • 17 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.
no code implementations • 9 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}$.
no code implementations • 8 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.
no code implementations • 3 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.
no code implementations • 1 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).
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).
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.
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.
no code implementations • 1 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.
no code implementations • 30 Dec 2020 • Haozhe Jia, Weidong Cai, Heng Huang, Yong Xia
In this paper, we propose a Hybrid High-resolution and Non-local Feature Network (H2NF-Net) to segment brain tumor in multimodal MR images.
no code implementations • 10 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.
no code implementations • 15 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.
1 code implementation • 11 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.
no code implementations • 1 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.
no code implementations • 24 Aug 2020 • Hongchang Gao, Heng Huang
To the best of our knowledge, this is the first adaptive decentralized training approach.
no code implementations • 24 Aug 2020 • Hongchang Gao, Heng Huang
The condition for achieving the linear speedup is also provided for this variant.
no code implementations • 18 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.
no code implementations • 14 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.
no code implementations • 14 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.
no code implementations • 13 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.
no code implementations • 4 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}})$.
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.
1 code implementation • 1 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.
1 code implementation • 29 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.
no code implementations • 17 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.
1 code implementation • CVPR 2020 • Dongnan Liu, Donghao Zhang, Yang song, Fan Zhang, Lauren O'Donnell, Heng Huang, Mei Chen, Weidong Cai
More specifically, we first propose a nuclei inpainting mechanism to remove the auxiliary generated objects in the synthesized images.
no code implementations • 11 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.
no code implementations • 25 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.
1 code implementation • 15 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.
1 code implementation • 4 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.
no code implementations • 24 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