Task-Agnostic Vision Transformer for Distributed Learning of Image Processing. 2022

Boah Kim, and Jeongsol Kim, and Jong Chul Ye

Recently, distributed learning approaches have been studied for using data from multiple sources without sharing them, but they are not usually suitable in applications where each client carries out different tasks. Meanwhile, Transformer has been widely explored in computer vision area due to its capability to learn the common representation through global attention. By leveraging the advantages of Transformer, here we present a new distributed learning framework for multiple image processing tasks, allowing clients to learn distinct tasks with their local data. This arises from a disentangled representation of local and non-local features using a task-specific head/tail and a task-agnostic Vision Transformer. Each client learns a translation from its own task to a common representation using the task-specific networks, while the Transformer body on the server learns global attention between the features embedded in the representation. To enable decomposition between the task-specific and common representations, we propose an alternating training strategy between clients and server. Experimental results on distributed learning for various tasks show that our method synergistically improves the performance of each client with its own data.

UI MeSH Term Description Entries

Related Publications

Boah Kim, and Jeongsol Kim, and Jong Chul Ye
July 2023, IEEE transactions on medical imaging,
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
July 2023, Journal of imaging,
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
May 2023, Neural networks : the official journal of the International Neural Network Society,
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
December 2022, IEEE transactions on visualization and computer graphics,
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
May 2022, Medical image analysis,
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
January 2023, Diagnostics (Basel, Switzerland),
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
May 2023, Computers in biology and medicine,
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
November 2023, IEEE transactions on medical imaging,
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
July 2022, Annual International Conference of the IEEE Engineering in Medicine and Biology Society. IEEE Engineering in Medicine and Biology Society. Annual International Conference,
Boah Kim, and Jeongsol Kim, and Jong Chul Ye
October 2022, Medical image analysis,
Copied contents to your clipboard!