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

Yong Zhang 0004

dblp:66/4615-4 · DBLP profile ↗
← Back
49ranked-venue papers
4as first author
35since 2021 · last 2026
0000-0002-0238-0719ORCID · conflict

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

Artificial intelligence and machine learning · 30 · 1 first-author · 28 since 2021Graphics, computer vision, multimedia, augmented reality and games · 27 · 2 first-author · 18 since 2021Applied, interdisciplinary, general and emerging computing · 11 · 3 first-authorDatabases, data management, data science and information retrieval · 9 · 9 since 2021Systems, architecture and hardware · 1 · 1 since 2021

Expertise — from the expertise taxonomy: the topics of the expert's papers under the CCF categories. A weight counts papers with recency: 1 for a paper about the topic, 0.3 when the topic is its context, halved every five years.

Artificial intelligence
18 papers
Efficient and distributed learning · 34% Trustworthy machine learning · 28% Image recognition and object detection · 11%
Theoretical computer science
7 papers
Mathematical optimization · 96% Algorithmic game theory and mechanism design · 4%
Network and information security
5 papers
Security and privacy of machine learning · 49% Digital forensics and information hiding · 26% Blockchain and cryptocurrency security · 20%
Computer architecture, parallel and distributed computing, and storage systems
2 papers
Cloud and datacenter computing · 100%

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

TopicWeightPapersLastEvidence papers
Machine learning › Efficient and distributed learning
federated learning
2.342024
Fair and Efficient Contribution Valuation for Vertical Federated Learning · ICLR 2024
Improving Fairness for Data Valuation in Horizontal Federated Learning · ICDE 2022
Towards Fair Federated Learning · KDD 2021
Machine learning › Trustworthy machine learning
fairness
1.122022
Improving Fairness for Data Valuation in Horizontal Federated Learning · ICDE 2022
Towards Fair Federated Learning · KDD 2021
Machine learning › Trustworthy machine learning
interpretability
1.022021
Robust Counterfactual Explanations on Graph Neural Networks · NeurIPS 2021
Finding Representative Interpretations on Convolutional Neural Networks · ICCV 2021
Machine learning › Graph learning
graph neural network
1.032025
Robust Counterfactual Explanations on Graph Neural Networks · NeurIPS 2021
Learn2Aggregate: Supervised Generation of Chvatal-Gomory Cuts Using Graph Neural Networks · AAAI 2025
Smart Initial Basis Selection for Linear Programs · ICML 2023
Digital forensics and information hiding
watermarking
0.922024
NFT-Based Data Marketplace with Digital Watermarking · KDD 2023
LaWa: Using Latent Space for In-Generation Image Watermarking · ECCV (89) 2024
Natural language and speech › Language models and text generation › decoding
contrastive decoding
0.912025
Enhancing Learned Knowledge in LoRA Adapters Through Efficient Contrastive Decoding on Ascend NPUs · KDD (2) 2025
Natural language and speech › Language models and text generation › decoding
decoding strategy
0.912025
Enhancing Learned Knowledge in LoRA Adapters Through Efficient Contrastive Decoding on Ascend NPUs · KDD (2) 2025
Natural language and speech › Question answering and dialogue systems
domain-specific question answering
0.912025
Evaluating LLM Reasoning in the Operations Research Domain with ORQA · AAAI 2025
Machine learning › Efficient and distributed learning
inference serving
0.912025
Efficiently Serving Large Multimodal Models Using EPD Disaggregation · ICML 2025
Cloud and datacenter computing
inference serving
0.912025
Efficiently Serving Large Multimodal Models Using EPD Disaggregation · ICML 2025
Mathematical optimization › integer programming › cutting planes
chvátal-gomory cuts
0.912025
Learn2Aggregate: Supervised Generation of Chvatal-Gomory Cuts Using Graph Neural Networks · AAAI 2025
Mathematical optimization › discrete optimization › mixed integer linear programming
cut generation
0.912025
Learn2Aggregate: Supervised Generation of Chvatal-Gomory Cuts Using Graph Neural Networks · AAAI 2025
Mathematical optimization › integer programming
cutting planes
0.912025
Learn2Aggregate: Supervised Generation of Chvatal-Gomory Cuts Using Graph Neural Networks · AAAI 2025
Mathematical optimization
integer programming
0.912025
Learn2Aggregate: Supervised Generation of Chvatal-Gomory Cuts Using Graph Neural Networks · AAAI 2025
Mathematical optimization › optimization modeling
LLM-based optimization modeling
0.912025
Evaluating LLM Reasoning in the Operations Research Domain with ORQA · AAAI 2025
Mathematical optimization › discrete optimization
mixed integer linear programming
0.912025
Learn2Aggregate: Supervised Generation of Chvatal-Gomory Cuts Using Graph Neural Networks · AAAI 2025
Mathematical optimization
optimization modeling
0.912025
Evaluating LLM Reasoning in the Operations Research Domain with ORQA · AAAI 2025
Machine learning › Trustworthy machine learning › robustness
adversarial attack
0.812024
Cocktail Universal Adversarial Attack on Deep Neural Networks · ECCV (65) 2024
Machine learning › Efficient and distributed learning › federated learning › federated learning architecture
vertical federated learning
0.812024
Fair and Efficient Contribution Valuation for Vertical Federated Learning · ICLR 2024
Image and video processing › information hiding
image watermarking
0.812024
LaWa: Using Latent Space for In-Generation Image Watermarking · ECCV (89) 2024
Computer vision › Image recognition and object detection
object detection
0.712023
ETran: Energy-Based Transferability Estimation · ICCV 2023
Machine learning › Transfer learning and domain adaptation
transferability estimation
0.712023
ETran: Energy-Based Transferability Estimation · ICCV 2023
Mathematical optimization › large-scale optimization › decomposition methods
column generation
0.712023
Smart Initial Basis Selection for Linear Programs · ICML 2023
Mathematical optimization
linear programming
0.712023
Smart Initial Basis Selection for Linear Programs · ICML 2023
Mathematical optimization › linear programming
simplex method
0.712023
Smart Initial Basis Selection for Linear Programs · ICML 2023
Machine learning › Efficient and distributed learning › adaptive computation
adaptive inference
0.612022
E-LANG: Energy-Based Joint Inferencing of Super and Swift Language Models · ACL (1) 2022
Computer vision › Image recognition and object detection › object detection
data augmentation for detection
0.612022
SemAug: Semantically Meaningful Image Augmentations for Object Detection Through Language Grounding · ECCV (36) 2022
Machine learning › Efficient and distributed learning › inference efficiency
LLM inference optimization
0.612022
E-LANG: Energy-Based Joint Inferencing of Super and Swift Language Models · ACL (1) 2022
Machine learning › Efficient and distributed learning
model compression
0.612022
E-LANG: Energy-Based Joint Inferencing of Super and Swift Language Models · ACL (1) 2022
Security and privacy of machine learning
model stealing
0.612022
Cosine Model Watermarking against Ensemble Distillation · AAAI 2022

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

graph neural network · 3.1shapley value · 2.7supervised learning · 1.7role switching · 1.7feature engineering · 1.7expert annotation · 1.7encode-prefill-decode disaggregation · 1.7benchmark construction · 1.7KV cache management · 1.7latent space representation · 1.5watermarking · 1.3error correction codes · 1.3NFT · 1.3latency optimization · 1.0deep reinforcement learning · 1.0deep neural network splitting · 1.0contrastive decoding · 0.9cooperative game theory · 0.8
YearPublicationVenuePosition
2026 Blind Watermarking for Tabular Datasets in Machine Learning: A Primary Key-Free Method
abstract
Watermarking is widely used to protect the ownership of tabular datasets. Comparing to non-blind watermarking that requires the original dataset to detect watermark, blind watermarking is more secure because it can accurately detect watermark without using the original dataset, which restricts public access to the original dataset. Existing blind watermarking methods rely on either a primary key or a virtual primary key to watermark a tabular dataset. However, these watermarks can be easily removed by an attacker with little to no impact on the dataset’s machine learning utility, because a primary key can be significantly modified without affecting the machine learning utility, and a virtual primary key is fragile to slight modifications on the dataset. Can we design a blind watermarking method without relying on a primary key or virtual primary key? In this article, we tackle this challenging task by a novel primary key-free method that embeds a sinusoidal signal as the watermark into a discrete-time signal constructed from the tabular dataset. We theoretically analyzed the robustness of our watermark against six challenging attacks, and empirically validated the outstanding performance of our method through comprehensive experiments on two real-world datasets.
Shaoxin Li 0002, David (Ming Xuan) Yue, Yong Zhang 0004, Lingyang Chu
ACM Trans. Knowl. Discov. Data5
2025 Learn2Aggregate: Supervised Generation of Chvatal-Gomory Cuts Using Graph Neural Networks
abstract
We present Learn2Aggregate, a machine learning (ML) framework for optimizing the generation of Chvatal-Gomory (CG) cuts in mixed integer linear programming (MILP). The framework trains a graph neural network to classify useful constraints for aggregation in CG cut generation. The ML-driven CG separator selectively focuses on a small set of impactful constraints, improving runtimes without compromising the strength of the generated cuts. Key to our approach is the formulation of a constraint classification task which favours sparse aggregation of constraints, consistent with empirical findings. This, in conjunction with a careful constraint labeling scheme and a hybrid of deep learning and feature engineering, results in enhanced CG cut generation across five diverse MILP benchmarks. On the largest test sets, our method closes roughly twice as much of the integrality gap as the standard CG method while running 40% faster. This performance improvement is due to our method eliminating 75% of the constraints prior to aggregation.
Arnaud Deza, Elias B. Khalil, Zhenan Fan, Zirui Zhou, Yong Zhang 0004
AAAI5
2025 Evaluating LLM Reasoning in the Operations Research Domain with ORQA
abstract
In this paper, we introduce and apply Operations Research Question Answering (ORQA), a new benchmark, to assess the generalization capabilities of Large Language Models (LLMs) in the specialized technical domain of Operations Research (OR). This benchmark is designed to evaluate whether LLMs can emulate the knowledge and reasoning skills of OR experts when given diverse and complex optimization problems. The dataset, crafted by OR experts, presents real-world optimization problems that require multistep reasoning to build their mathematical models. Our evaluations of various open-source LLMs, such as LLaMA 3.1, DeepSeek, and Mixtral reveal their modest performance, indicating a gap in their aptitude to generalize to specialized technical domains. This work contributes to the ongoing discourse on LLMs’ generalization capabilities, providing insights for future research in this area. The dataset and evaluation code are publicly available.
Mahdi Mostajabdaveh, Timothy T. L. Yu, Samarendra Chandan Bindu Dash, Rindranirina Ramamonjison, Jabo Serge Byusa, Giuseppe Carenini, Zirui Zhou, Yong Zhang 0004
AAAI8
2025 Efficiently Serving Large Multimodal Models Using EPD Disaggregation
abstract
Large Multimodal Models (LMMs) extend Large Language Models (LLMs) by handling diverse inputs such as images, audio, and video, but at the cost of adding a multimodal encoding stage that increases both computational and memory overhead. This step negatively affects key Service Level Objectives (SLOs), such as time to first token (TTFT) and time per output token (TPOT). We introduce Encode-Prefill-Decode (EPD) Disaggregation, a novel framework that separates the encoding, prefill, and decode stages onto dedicated resources. Unlike current systems, which bundle encoding and prefill together, our approach decouples these steps, unlocking new opportunities and optimizations. These include a mechanism to cache multimedia tokens for efficient transfer, a novel way to parallelize the encoding load within a request, a module for optimal resource allocation for disaggregated serving, and a novel role-switching method to handle changing workload characteristics. Experimental evaluations with popular LMMs show substantial gains in memory efficiency (up to 15$\times$ lower peak memory utilization), batch sizes (up to 22$\times$ larger), 10$\times$ more images per request, and 2.2$\times$ larger KV caches. Furthermore, it leads to significant improvements in SLO attainment (up to 90–100% improvement) and TTFT (up to 71% reduction), compared to systems that do not disaggregate. The code is available at https://github.com/vbdi/epdserve.
Gursimran Singh, Xinglu Wang, Timothy T. L. Yu, Linzi Xing, Xiaolong Bai, Yong Zhang 0004, Zhenan Fan
ICML11
2025 Enhancing Learned Knowledge in LoRA Adapters Through Efficient Contrastive Decoding on Ascend NPUs
abstract
Huawei Cloud users leverage LoRA (Low-Rank Adaptation) as an efficient and scalable method to fine-tune and customize large language models (LLMs) for application-specific needs. However, tasks that require complex reasoning or deep contextual understanding are often hindered by biases or interference from the base model when using typical decoding methods like greedy or beam search. These biases can lead to generic or task-agnostic responses from the base model instead of leveraging the LoRA-specific adaptations. In this paper, we introduce Contrastive LoRA Decoding (CoLD), a novel decoding framework designed to maximize the use of task-specific knowledge in LoRA-adapted models, resulting in better downstream performance. CoLD uses contrastive decoding by scoring candidate tokens based on the divergence between the probability distributions of a LoRA-adapted expert model and the corresponding base model. This approach prioritizes tokens that better align with the LoRA's learned representations, enhancing performance for specialized tasks. While effective, a naive implementation of CoLD is computationally expensive because each decoding step requires evaluating multiple token candidates across both models. To address this, we developed an optimized kernel for Huawei's Ascend NPU. CoLD achieves up to a 5.54% increase in task accuracy while reducing end-to-end latency by 28% compared to greedy decoding. This work provides practical and efficient decoding strategies for fine-tuned LLMs in resource-constrained environments and has broad implications for applied data science in both cloud and on-premises settings.
Morgan Lindsay Heisler, Linzi Xing, Hanieh Sadri, Gursimran Singh, Yong Zhang 0004, Zhenan Fan
KDD (2)9
2025 Towards Secure and Usable 3D Assets: A Novel Framework for Automatic Visible Watermarking
abstract
3D models, particularly AI-generated ones, have wit-nessed a recent surge across various industries such as en-tertainment. Hence, there is an alarming need to protect the intellectual property and avoid the misuse of these valuable assets. As a viable solution to address these concerns, we rigorously define the novel task of automated 3D visible wa-termarking in terms of two competing aspects: watermark quality and asset utility. Moreover, we propose a method of embedding visible watermarks that automatically deter-mines the right location, orientation, and number of wa-termarks to be placed on arbitrary 3D assets for high wa-termark quality and asset utility. Our method is based on a novel rigid-body optimization that uses back-propagation to automatically learn transforms for ideal watermark place-ment. In addition, we propose a novel curvature-matching method for fusing the watermark into the 3D model that further improves readability and security. Finally, we provide a detailed experimental analysis on two benchmark 3D datasets validating the superior performance of our approach in comparison to baselines. Code and demo are available here11https://developer.huaweicloud.com/develop/aigallery/notebook/detail?id=15adbaaa-2583-4ec3-804a-61c29f001e03.
Gursimran Singh, Tianxi Hu, Qiang Tang 0002, Yong Zhang 0004
WACV5
2024 Towards Human-aligned Evaluation for Linear Programming Word Problems
abstract
Math Word Problem (MWP) is a crucial NLP task aimed at providing solutions for given mathematical descriptions. A notable sub-category of MWP is the Linear Programming Word Problem (LPWP), which holds significant relevance in real-world decision-making and operations research. While the recent rise of generative large language models (LLMs) has brought more advanced solutions to LPWPs, existing evaluation methodologies for this task still diverge from human judgment and face challenges in recognizing mathematically equivalent answers. In this paper, we introduce a novel evaluation metric rooted in graph edit distance, featuring benefits such as permutation invariance and more accurate program equivalence identification. Human evaluations empirically validate the superior efficacy of our proposed metric when particularly assessing LLM-based solutions for LPWP.
Linzi Xing, Xinglu Wang, Yuxi Feng, Zhenan Fan, Zhijiang Guo, Xiaojin Fu, Rindranirina Ramamonjison, Mahdi Mostajabdaveh, Xiongwei Han, Zirui Zhou, Yong Zhang 0004
LREC/COLING12
2024 Cocktail Universal Adversarial Attack on Deep Neural Networks
Shaoxin Li 0002, Xiaofeng Liao 0001, Yong Zhang 0004, Lingyang Chu
ECCV (65)5
2024 LaWa: Using Latent Space for In-Generation Image Watermarking
Ahmad Rezaei, Saeed Ranjbar Alvar, Arezou Fatemi, Yong Zhang 0004
ECCV (89)5
2024 Fair and Efficient Contribution Valuation for Vertical Federated Learning
abstract
Federated learning is an emerging technology for training machine learning models across decentralized data sources without sharing data. Vertical federated learning, also known as feature-based federated learning, applies to scenarios where data sources have the same sample IDs but different feature sets. To ensure fairness among data owners, it is critical to objectively assess the contributions from different data sources and compensate the corresponding data owners accordingly. The Shapley value is a provably fair contribution valuation metric originating from cooperative game theory. However, its straight-forward computation requires extensively retraining a model on each potential combination of data sources, leading to prohibitively high communication and computation overheads due to multiple rounds of federated learning. To tackle this challenge, we propose a contribution valuation metric called vertical federated Shapley value (VerFedSV) based on the classic Shapley value. We show that VerFedSV not only satisfies many desirable properties of fairness but is also efficient to compute. Moreover, VerFedSV can be adapted to both synchronous and asynchronous vertical federated learning algorithms. Both theoretical analysis and extensive experimental results demonstrate the fairness, efficiency, adaptability, and effectiveness of VerFedSV.
Zhenan Fan, Huang Fang, Xinglu Wang, Zirui Zhou, Jian Pei 0001, Michael P. Friedlander, Yong Zhang 0004
ICLR7
2024 Primary Key Free Watermarking for Numerical Tabular Datasets in Machine Learning
Shaoxin Li 0002, David (Ming Xuan) Yue, Yong Zhang 0004, Lingyang Chu
ICPR (31)5
2023 ArchBERT: Bi-Modal Understanding of Neural Architectures and Natural Languages
abstract
Building multi-modal language models has been a trend in the recent years, where additional modalities such as image, video, speech, etc. are jointly learned along with natural languages (i.e., textual information).Despite the success of these multi-modal language models with different modalities, there is no existing solution for neural network architectures and natural languages.Providing neural architectural information as a new modality allows us to provide fast architecture-2-text and text-2-architecture retrieval/generation services on the cloud with a single inference.Such solution is valuable in terms of helping beginner and intermediate ML users to come up with better neural architectures or AutoML approaches with a simple text query.In this paper, we propose ArchBERT, a bi-modal model for joint learning and understanding of neural architectures and natural languages, which opens up new avenues for research in this area.We also introduce a pre-training strategy named Masked Architecture Modeling (MAM) for a more generalized joint learning.Moreover, we introduce and publicly release two new bi-modal datasets for training and validating our methods.The ArchBERT's performance is verified through a set of numerical experiments on different downstream tasks such as architecture-oriented reasoning, question answering, and captioning (summarization).Datasets, codes, and demos are available as supplementary materials 1 .
Saeed Ranjbar Alvar, Behnam Kamranian, Amin Banitalebi-Dehkordi, Yong Zhang 0004
CoNLL5
2023 ETran: Energy-Based Transferability Estimation
abstract
This paper addresses the problem of ranking pre-trained models for object detection and image classification. Selecting the best pre-trained model by fine-tuning is an expensive and time-consuming task. Previous works have proposed transferability estimation based on features extracted by the pre-trained models. We argue that quantifying whether the target dataset is in-distribution (IND) or out-of-distribution (OOD) for the pre-trained model is an important factor in the transferability estimation. To this end, we propose ETran, an energy-based transferability assessment metric, which includes three scores: 1) energy score, 2) classification score, and 3) regression score. We use energy-based models to determine whether the target dataset is OOD or IND for the pre-trained model. In contrast to the prior works, ETran is applicable to a wide range of tasks including classification, regression, and object detection (classification+regression). This is the first work that proposes transferability estimation for object detection task. Our extensive experiments on four benchmarks and two tasks show that ETran outperforms previous works on object detection and classification benchmarks by an average of 21% and 12%, respectively, and achieves SOTA in transferability assessment. Code is available here1.
Mohsen Gholami, Xinglu Wang, Behnam Kamranian, Yong Zhang 0004
ICCV5
2023 Smart Initial Basis Selection for Linear Programs
abstract
The simplex method, introduced by Dantzig more than half a century ago, is still to date one of the most efficient methods for solving large-scale linear programming (LP) problems. While the simplex method is known to have the finite termination property under mild assumptions, the number of iterations until optimality largely depends on the choice of initial basis. Existing strategies for selecting an advanced initial basis are mostly rule-based. These rules usually require extensive expert knowledge and empirical study to develop. Yet, many of them fail to exhibit consistent improvement, even for LP problems that arise in a single application scenario. In this paper, we propose a learning-based approach for initial basis selection. We employ graph neural networks as a building block and develop a model that attempts to capture the relationship between LP problems and their optimal bases. In addition, during the inference phase, we supplement the learning-based prediction with linear algebra tricks to ensure the validity of the generated initial basis. We validate the effectiveness of our proposed strategy by extensively testing it with state-of-the-art simplex solvers, including the open-source solver HiGHS and the commercial solver OptVerse. Through these rigorous experiments, we demonstrate that our strategy achieves substantial speedup and consistently outperforms existing rule-based methods. Furthermore, we extend the proposed approach to generating restricted master problems for column generation methods and present encouraging numerical results.
Zhenan Fan, Xinglu Wang, Oleksandr Yakovenko, Abdullah Ali Sivas, Owen Ren, Yong Zhang 0004, Zirui Zhou
ICML6
2023 Asynchronous, Option-Based Multi-Agent Policy Gradient: A Conditional Reasoning Approach
abstract
Cooperative multi-agent problems often require coordination between agents, which can be achieved through a centralized policy that considers the global state. Multi-agent policy gradient (MAPG) methods are commonly used to learn such policies, but they are often limited to problems with low-level action spaces. In complex problems with large state and action spaces, it is advantageous to extend MAPG methods to use higher-level actions, also known as options, to improve the policy search efficiency. However, multi-robot option executions are often asynchronous, that is, agents may select and complete their options at different time steps. This makes it difficult for MAPG methods to derive a centralized policy and evaluate its gradient, as centralized policy always select new options at the same time. In this work, we propose a novel, conditional reasoning approach to address this problem and demonstrate its effectiveness on representative option-based multi-agent cooperative tasks through empirical validation. Find code and videos at: https://sites.google.com/view/mahrlsupp/
Xubo Lyu, Amin Banitalebi-Dehkordi, Mo Chen 0001, Yong Zhang 0004
IROS4
2023 NFT-Based Data Marketplace with Digital Watermarking
abstract
In today's digital world, enterprises and individuals are generating massive data that is potentially useful for many data consumers with data driven applications. The emergence of data marketplaces is a step toward helping the data owners to monetize their digital assets and get connected to the potential buyers. The current data marketplaces cannot handle the challenges related to data ownership claims, illegal redistribution, and data ownership traceability. To overcome these problems in a general-purpose market, we propose a marketplace based on watermarking and Non-Fungible Token (NFT) technologies. In the proposed NFT-based marketplace, the owner's data is stored as an NFT where the underlying content of the NFT holds the watermarked data. The watermarked data is obtained by embedding some information about the owners and the buyers into the original data. The embedded information can later be extracted to identify the owner and the buyer of the traded data. Furthermore, the transactions corresponding to the NFT provide verifiable ownership proof and traceable ownership history. A Proof-Of-Concept (POC) implementation of the proposed marketplace that will be integrated within AI-Gallery Data Marketplace service in Huawei Cloud is presented for trading image data. An extensive set of experiments to measure the gas consumption on the blockchain and evaluate the robustness of the watermarked assets against 51 attacks are performed. Finally, a method based on error correction codes is proposed for improving the watermarking robustness in the implemented marketplace. The link for the codes and the POC demo is provided in the appendix.
Saeed Ranjbar Alvar, David (Ming Xuan) Yue, Yong Zhang 0004
KDD4
2023 Exact Combinatorial Optimization with Temporo-Attentional Graph Neural Networks
Mehdi Seyfi, Amin Banitalebi-Dehkordi, Zirui Zhou, Yong Zhang 0004
ECML/PKDD (4)4
2022 Cosine Model Watermarking against Ensemble Distillation
abstract
Many model watermarking methods have been developed to prevent valuable deployed commercial models from being stealthily stolen by model distillations. However, watermarks produced by most existing model watermarking methods can be easily evaded by ensemble distillation, because averaging the outputs of multiple ensembled models can significantly reduce or even erase the watermarks. In this paper, we focus on tackling the challenging task of defending against ensemble distillation. We propose a novel watermarking technique named CosWM to achieve outstanding model watermarking performance against ensemble distillation. CosWM is not only elegant in design, but also comes with desirable theoretical guarantees. Our extensive experiments on public data sets demonstrate the excellent performance of CosWM and its advantages over the state-of-the-art baselines.
Laurent Charette, Lingyang Chu, Jian Pei 0001, Lanjun Wang, Yong Zhang 0004
AAAI6
2022 E-LANG: Energy-Based Joint Inferencing of Super and Swift Language Models
abstract
Building huge and highly capable language models has been a trend in the past years.Despite their great performance, they incur high computational cost.A common solution is to apply model compression or choose light-weight architectures, which often need a separate fixed-size model for each desirable computational budget, and may lose performance in case of heavy compression.This paper proposes an effective dynamic inference approach, called E-LANG, which distributes the inference between large accurate Supermodels and light-weight Swift models.To this end, a decision making module routes the inputs to Super or Swift models based on the energy characteristics of the representations in the latent space.This method is easily adoptable and architecture agnostic.As such, it can be applied to black-box pre-trained models without a need for architectural manipulations, reassembling of modules, or re-training.Unlike existing methods that are only applicable to encoder-only backbones and classification tasks, our method also works for encoderdecoder structures and sequence-to-sequence tasks such as translation.The E-LANG performance is verified through a set of experiments with T5 and BERT backbones on GLUE, Su-perGLUE, and WMT.In particular, we outperform T5-11B with an average computations speed-up of 3.3× on GLUE and 2.9× on SuperGLUE.We also achieve BERT-based SOTA on GLUE with 3.2× less computations.Code and demo are available here.
Amin Banitalebi-Dehkordi, Yong Zhang 0004
ACL (1)3
2022 SemAug: Semantically Meaningful Image Augmentations for Object Detection Through Language Grounding
Morgan Lindsay Heisler, Amin Banitalebi-Dehkordi, Yong Zhang 0004
ECCV (36)3
2022 Improving Fairness for Data Valuation in Horizontal Federated Learning
abstract
Federated learning is an emerging decentralized machine learning scheme that allows multiple data owners to work collaboratively while ensuring data privacy. The success of federated learning depends largely on the participation of data owners. To sustain and encourage data owners' participation, it is crucial to fairly evaluate the quality of the data provided by the data owners as well as their contribution to the final model and reward them correspondingly. Federated Shapley value, recently proposed by Wang et al. [Federated Learning, 2020], is a measure for data value under the framework of federated learning that satisfies many desired properties for data valuation. However, there are still factors of potential unfairness in the design of federated Shapley value because two data owners with the same local data may not receive the same evaluation. We propose a new measure called completed federated Shapley value to improve the fairness of federated Shapley value. The design depends on completing a matrix consisting of all the possible contributions by different subsets of the data owners. It is shown under mild conditions that this matrix is approximately low-rank by leveraging concepts and tools from optimization. Both theoretical analysis and empirical evaluation verify that the proposed measure does improve fairness in many circumstances.
Zhenan Fan, Huang Fang, Zirui Zhou, Jian Pei 0001, Michael P. Friedlander, Changxin Liu 0001, Yong Zhang 0004
ICDE7
2022 Deep Reinforcement Learning for Exact Combinatorial Optimization: Learning to Branch
abstract
Branch-and-bound is a systematic enumerative method for combinatorial optimization, where the performance highly relies on the variable selection strategy. State-of-the-art handcrafted heuristic strategies suffer from relatively slow inference time for each selection, while the current machine learning methods require a significant amount of labeled data. We propose a new approach for solving the data labeling and inference latency issues in combinatorial optimization based on the use of the reinforcement learning (RL) paradigm. We use imitation learning to bootstrap an RL agent and then use Proximal Policy Optimization (PPO) to further explore global optimal actions. Then, a value network is used to run Monte-Carlo tree search (MCTS) to enhance the policy network. We evaluate the performance of our method on four different categories of combinatorial optimization problems and show that our approach performs strongly compared to the state-of-the-art machine learning and heuristics based methods.
Tianyu Zhang 0003, Amin Banitalebi-Dehkordi, Yong Zhang 0004
ICPR3
2022 Mining Minority-Class Examples with Uncertainty Estimates
Gursimran Singh, Lingyang Chu, Lanjun Wang, Jian Pei 0001, Yong Zhang 0004
MMM (1)6
2022 Data pricing in machine learning pipelines
Zicun Cong, Jian Pei 0001, Feida Zhu 0001, Yong Zhang 0004
Knowl. Inf. Syst.5
2022 Extending Momentum Contrast With Cross Similarity Consistency Regularization
abstract
Contrastive self-supervised representation learning methods maximize the similarity between the positive pairs, and at the same time tend to minimize the similarity between the negative pairs. However, in general the interplay between the negative pairs is ignored as they do not put in place special mechanisms to treat negative pairs differently according to their specific differences and similarities. In this paper, we present Extended Momentum Contrast (XMoCo), a self-supervised representation learning method founded upon the legacy of the momentum-encoder unit proposed in the MoCo family configurations. To this end, we introduce a cross consistency regularization loss, with which we extend the transformation consistency to dissimilar images (negative pairs). Under the cross consistency regularization rule, we argue that semantic representations associated with any pair of images (positive or negative) should preserve their cross-similarity under pretext transformations. Moreover, we further regularize the training loss by enforcing a uniform distribution of similarity over the negative pairs across a batch. The proposed regularization can easily be added to existing self-supervised learning algorithms in a plug-and-play fashion. Empirically, we report a competitive performance on the standard Imagenet-1K linear head classification benchmark. In addition, by transferring the learned representations to common downstream tasks, we show that using XMoCo with the prevalently utilized augmentations can lead to improvements in the performance of such tasks. We hope the findings of this paper serve as a motivation for researchers to take into consideration the important interplay among the negative examples in self-supervised learning.
Mehdi Seyfi, Amin Banitalebi-Dehkordi, Yong Zhang 0004
IEEE Trans. Circuits Syst. Video Technol.3
2021 Personalized Cross-Silo Federated Learning on Non-IID Data
abstract
Non-IID data present a tough challenge for federated learning. In this paper, we explore a novel idea of facilitating pairwise collaborations between clients with similar data. We propose FedAMP, a new method employing federated attentive message passing to facilitate similar clients to collaborate more. We establish the convergence of FedAMP for both convex and non-convex models, and propose a heuristic method to further improve the performance of FedAMP when clients adopt deep neural networks as personalized models. Our extensive experiments on benchmark data sets demonstrate the superior performance of the proposed methods.
Yutao Huang, Lingyang Chu, Zirui Zhou, Lanjun Wang, Jiangchuan Liu, Jian Pei 0001, Yong Zhang 0004
AAAI7
2021 EBJR: Energy-Based Joint Reasoning for Adaptive Inference
Amin Banitalebi-Dehkordi, Yong Zhang 0004
BMVC3
2021 Repaint: Improving the Generalization of Down-Stream Visual Tasks by Generating Multiple Instances of Training Examples
Amin Banitalebi-Dehkordi, Yong Zhang 0004
BMVC2
2021 Model Composition: Can Multiple Neural Networks Be Combined into a Single Network Using Only Unlabeled Data?
Amin Banitalebi-Dehkordi, Xinyu Kang, Yong Zhang 0004
BMVC3
2021 Finding Representative Interpretations on Convolutional Neural Networks
abstract
Interpreting the decision logic behind effective deep convolutional neural networks (CNN) on images complements the success of deep learning models. However, the existing methods can only interpret some specific decision logic on individual or a small number of images. To facilitate human understandability and generalization ability, it is important to develop representative interpretations that interpret common decision logics of a CNN on a large group of similar images, which reveal the common semantics data contributes to many closely related predictions. In this paper, we develop a novel unsupervised approach to produce a highly representative interpretation for a large number of similar images. We formulate the problem of finding representative interpretations as a co-clustering problem, and convert it into a submodular cost submodular cover problem based on a sample of the linear decision boundaries of a CNN. We also present a visualization and similarity ranking method. Our extensive experiments demonstrate the excellent performance of our method.
Peter Cho-Ho Lam, Lingyang Chu, Maxim Torgonskiy, Jian Pei 0001, Yong Zhang 0004, Lanjun Wang
ICCV5
2021 SimROD: A Simple Adaptation Method for Robust Object Detection
abstract
This paper presents a Simple and effective unsupervised adaptation method for Robust Object Detection (SimROD). To overcome the challenging issues of domain shift and pseudo-label noise, our method integrates a novel domain-centric data augmentation, a gradual self-labeling adaptation procedure, and a teacher-guided fine-tuning mechanism. Using our method, target domain samples can be leveraged to adapt object detection models without changing the model architecture or generating synthetic data. When applied to image corruptions and high-level cross-domain adaptation benchmarks, our method outperforms prior baselines on multiple domain adaptation benchmarks. SimROD achieves new state-of-the-art on standard real-to-synthetic and cross-camera setup benchmarks. On the image corruption benchmark, models adapted with our method achieved a relative robustness improvement of 15-25% AP50 on Pascal-C and 5-6% AP on COCO-C and Cityscapes-C. On the cross-domain benchmark, our method outperformed the best baseline performance by up to 8% and 4% AP50 on Comic and Watercolor respectively.1
Rindranirina Ramamonjison, Amin Banitalebi-Dehkordi, Xinyu Kang, Xiaolong Bai, Yong Zhang 0004
ICCV5
2021 Stealthy Targeted Data Poisoning Attack on Knowledge Graphs
abstract
A host of different KG embedding techniques have emerged recently and have been empirically shown to be very effective in accurately predicting missing facts in a KG, thus improving its coverage and quality. Unfortunately, embedding techniques can fall prey to adversarial data poisoning attack. In this form of attack, facts may be added to or deleted from a KG, called performing perturbations, that results in the manipulation of the plausibility of target facts in a KG. While recent works confirm this intuition, the attacks considered there ignore the risk of exposure. Intuitively, an attack is of limited value if it is highly likely to be caught, i.e., exposed. To address this, we introduce a notion of the exposure risk and propose a novel problem of attacking a KG by means of perturbations where the goal is to maximize the manipulation of the target fact's plausibility while keeping the risk of exposure under a given budget. We design a deep reinforcement learning-based framework, called RATA, that learns to use low-risk perturbations without compromising on the performance, i.e., manipulation of target fact plausibility. We test the performance of RATA against recently proposed strategies for KG attacks, on two different benchmark datasets and on different kinds of target facts. Our experiments show that RATA achieves state-of-the-art performance even while using a fraction of the risk.
Prithu Banerjee, Lingyang Chu, Yong Zhang 0004, Laks V. S. Lakshmanan, Lanjun Wang
ICDE3
2021 Auto-Split: A General Framework of Collaborative Edge-Cloud AI
abstract
In many industry scale applications, large and resource consuming machine learning models reside in powerful cloud servers. At the same time, large amounts of input data are collected at the edge of cloud. The inference results are also communicated to users or passed to downstream tasks at the edge. The edge often consists of a large number of low-power devices. It is a big challenge to design industry products to support sophisticated deep model deployment and conduct model inference in an efficient manner so that the model accuracy remains high and the end-to-end latency is kept low. This paper describes the techniques and engineering practice behind Auto-Split, an edge-cloud collaborative prototype of Huawei Cloud. This patented technology is already validated on selected applications, is on its way for broader systematic edge-cloud application integration, and is being made available for public use as an automated pipeline service for end-to-end cloud-edge collaborative intelligence deployment. To the best of our knowledge, there is no existing industry product that provides the capability of Deep Neural Network (DNN) splitting.
Amin Banitalebi-Dehkordi, Naveen Vedula, Jian Pei 0001, Lanjun Wang, Yong Zhang 0004
KDD6
2021 Towards Fair Federated Learning
abstract
Federated learning has become increasingly popular as it facilitates collaborative training of machine learning models among multiple clients while preserving their data privacy. In practice, one major challenge for federated learning is to achieve fairness in collaboration among the participating clients, because different clients' contributions to a model are usually far from equal due to various reasons. Besides, as machine learning models are deployed in more and more important applications, how to achieve model fairness, that is, to ensure that a trained model has no discrimination against sensitive attributes, has become another critical desiderata for federated learning. In this tutorial, we discuss formulations and methods such that collaborative fairness, model fairness, and privacy can be fully respected in federated learning. We review the existing efforts and the latest progress, and discuss a series of potential directions.
Zirui Zhou, Lingyang Chu, Changxin Liu 0001, Lanjun Wang, Jian Pei 0001, Yong Zhang 0004
KDD6
2021 Robust Counterfactual Explanations on Graph Neural Networks
abstract
Massive deployment of Graph Neural Networks (GNNs) in high-stake applications generates a strong demand for explanations that are robust to noise and align well with human intuition. Most existing methods generate explanations by identifying a subgraph of an input graph that has a strong correlation with the prediction. These explanations are not robust to noise because independently optimizing the correlation for a single input can easily overfit noise. Moreover, they are not counterfactual because removing an identified subgraph from an input graph does not necessarily change the prediction result. In this paper, we propose a novel method to generate robust counterfactual explanations on GNNs by explicitly modelling the common decision logic of GNNs on similar input graphs. Our explanations are naturally robust to noise because they are produced from the common decision boundaries of a GNN that govern the predictions of many similar input graphs. The explanations are also counterfactual because removing the set of edges identified by an explanation from the input graph changes the prediction significantly. Exhaustive experiments on many public datasets demonstrate the superior performance of our method.
Mohit Bajaj, Lingyang Chu, Zi Yu Xue, Jian Pei 0001, Lanjun Wang, Peter Cho-Ho Lam, Yong Zhang 0004
NeurIPS7
2020 Logistic Regression Confined by Cardinality-Constrained Sample and Feature Selection
abstract
Many vision-based applications rely on logistic regression for embedding classification within a probabilistic context, such as recognition in images and videos or identifying disease-specific image phenotypes from neuroimages. Logistic regression, however, often performs poorly when trained on data that is noisy, has irrelevant features, or when the samples are distributed across the classes in an imbalanced setting; a common occurrence in visual recognition tasks. To deal with those issues, researchers generally rely on adhoc regularization techniques or model a subset of these issues. We instead propose a mathematically sound logistic regression model that selects a subset of (relevant) features and (informative and balanced) set of samples during the training process. The model does so by applying cardinality constraints (via ℓ0-`norm' sparsity) on the features and samples. ℓ0defines sparsity in mathematical settings but in practice has mostly been approximated (e.g., via ℓ1or its variations) for computational simplicity. We prove that a local minimum to the non-convex optimization problems induced by cardinality constraints can be computed by combining block coordinate descent with penalty decomposition. On synthetic, image recognition, and neuroimaging datasets, we show that the accuracy of the method is higher than alternative methods and classifiers commonly used in the literature.
Ehsan Adeli-Mosabbeb, Dongjin Kwon, Yong Zhang 0004, Kilian M. Pohl
IEEE Trans. Pattern Anal. Mach. Intell.4
2019 XQ-SR: Joint x-q space super-resolution with application to infant diffusion MRI
Geng Chen 0001, Bin Dong 0001, Yong Zhang 0004, Weili Lin, Dinggang Shen, Pew-Thian Yap
Medical Image Anal.3
2019 Denoising of Diffusion MRI Data via Graph Framelet Matching in x-q Space
abstract
Diffusion magnetic resonance imaging (DMRI) suffers from lower signal-to-noise-ratio (SNR) due to MR signal attenuation associated with the motion of water molecules. To improve SNR, the non-local means (NLM) algorithm has demonstrated state-of-the-art performance in noise reduction. However, existing NLM algorithms do not take into account explicitly the fact that DMRI signal can vary significantly with local fiber orientations. Applying NLM naïvely can hence blur subtle structures and aggravate partial volume effects. To overcome this limitation, we improve NLM by performing neighborhood matching in non-flat domains and removing noise with information from both x -space (spatial domain) and q -space (wavevector domain). Specifically, we first encode the q -space sampling domain using a graph. We then perform graph framelet transforms to extract robust rotation-invariant features for each sampling point in x-q space. The resulting features are employed for robust neighborhood matching to locate recurrent information. Finally, we remove noise via an NLM framework. To adapt to the various types of noise in multi-coil MR imaging, we transform the signal before denoising so that it is Gaussian-distributed, allowing noise removal to be carried out in an unbiased manner. Our method is able to more effectively locate recurrent information in white matter structures with different orientations, avoiding the blurring effects caused by naïvely applying NLM. Experiments on synthetic, repetitively-acquired, and infant DMRI data demonstrate that our method is able to preserve subtle structures while effectively removing noise.
Geng Chen 0001, Bin Dong 0001, Yong Zhang 0004, Weili Lin, Dinggang Shen, Pew-Thian Yap
IEEE Trans. Medical Imaging3
2017 q-Space Upsampling Using x-q Space Regularization
Geng Chen 0001, Bin Dong 0001, Yong Zhang 0004, Dinggang Shen, Pew-Thian Yap
MICCAI (1)3
2017 Neighborhood Matching for Curved Domains with Application to Denoising in Diffusion MRI
Geng Chen 0001, Bin Dong 0001, Yong Zhang 0004, Dinggang Shen, Pew-Thian Yap
MICCAI (1)3
2017 Computing group cardinality constraint solutions for logistic regression problems
Yong Zhang 0004, Dongjin Kwon, Kilian M. Pohl
Medical Image Anal.1
2016 Tight Graph Framelets for Sparse Diffusion MRI q-Space Representation
abstract
In diffusion MRI, the outcome of estimation problems can often be improved by taking into account the correlation of diffusion-weighted images scanned with neighboring wavevectors in q -space. For this purpose, we propose in this paper to employ tight wavelet frames constructed on non-flat domains for multi-scale sparse representation of diffusion signals. This representation is well suited for signals sampled regularly or irregularly, such as on a grid or on multiple shells, in q -space. Using spectral graph theory, the frames are constructed based on quasi-affine systems (i.e., generalized dilations and shifts of a finite collection of wavelet functions) defined on graphs, which can be seen as a discrete representation of manifolds. The associated wavelet analysis and synthesis transforms can be computed efficiently and accurately without the need for explicit eigen-decomposition of the graph Laplacian, allowing scalability to very large problems. We demonstrate the effectiveness of this representation, generated using what we call tight graph framelets , in two specific applications: denoising and super-resolution in q -space using \(\ell _{0}\) regularization. The associated optimization problem involves only thresholding and solving a trivial inverse problem in an iterative manner. The effectiveness of graph framelets is confirmed via evaluation using synthetic data with noncentral chi noise and real data with repeated scans. These keywords were added by machine and not by the authors. This process is experimental and the keywords may be updated as the learning algorithm improves.
Pew-Thian Yap, Bin Dong 0001, Yong Zhang 0004, Dinggang Shen
MICCAI (3)3
2016 Joint Data Harmonization and Group Cardinality Constrained Classification
Yong Zhang 0004, Sanghyun Park 0004, Kilian M. Pohl
MICCAI (1)1
2016 Multi-Tissue Decomposition of Diffusion MRI Signals via ℓ0 Sparse-Group Estimation
abstract
Sparse estimation techniques are widely utilized in diffusion magnetic resonance imaging (DMRI). In this paper, we present an algorithm for solving the ℓ0sparse-group estimation problem and apply it to the tissue signal separation problem in DMRI. Our algorithm solves the ℓ0problem directly, unlike existing approaches that often seek to solve its relaxed approximations. We include the mathematical proofs showing that the algorithm will converge to a solution satisfying the firstorder optimality condition within a finite number of iterations. We apply this algorithm to DMRI data to tease apart signal contributions from white matter, gray matter, and cerebrospinal fluid with the aim of improving the estimation of the fiber orientation distribution function (FODF). Unlike spherical deconvolution approaches that assume an invariant fiber response function (RF), our approach utilizes an RF group to span the signal subspace of each tissue type, allowing greater flexibility in accounting for possible variations of the RF throughout space and within each voxel. Our ℓ0algorithm allows for the natural groupings of the RFs to be considered during signal decomposition. Experimental results confirm that our method yields estimates of FODFs and volume fractions of tissue compartments with improved robustness and accuracy. Our ℓ0algorithm is general and can be applied to sparse estimation problems beyond the scope of this paper.
Pew-Thian Yap, Yong Zhang 0004, Dinggang Shen
IEEE Trans. Image Process.2
2015 Diffusion Compartmentalization Using Response Function Groups with Cardinality Penalization
Pew-Thian Yap, Yong Zhang 0004, Dinggang Shen
MICCAI (1)2
2015 Iterative Subspace Screening for Rapid Sparse Estimation of Brain Tissue Microstructural Properties
Pew-Thian Yap, Yong Zhang 0004, Dinggang Shen
MICCAI (1)2
2015 Brain Tissue Segmentation Based on Diffusion MRI Using ℓ0 Sparse-Group Representation Classification
Pew-Thian Yap, Yong Zhang 0004, Dinggang Shen
MICCAI (3)2
2015 Solving Logistic Regression with Group Cardinality Constraints for Time Series Analysis
Yong Zhang 0004, Kilian M. Pohl
MICCAI (3)1
2011 Penalty Decomposition Methods for Rank Minimization
abstract
In this paper we consider general rank minimization problems with rank appearing in either objective function or constraint. We first show that a class of matrix optimization problems can be solved as lower dimensional vector optimization problems. As a consequence, we establish that a class of rank minimization problems have closed form solutions. Using this result, we then propose penalty decomposition methods for general rank minimization problems. The convergence results of the PD methods have been shown in the longer version of the paper. Finally, we test the performance of our methods by applying them to matrix completion and nearest low-rank correlation matrix problems. The computational results demonstrate that our methods generally outperform the existing methods in terms of solution quality and/or speed.
Yong Zhang 0004, Zhaosong Lu
NIPS1