Saved in:
Bibliographic Details
Main Authors: Tian, Xinyu, Shen, Xiaotong
Format: Preprint
Published: 2024
Subjects:
Online Access:https://arxiv.org/abs/2405.16837
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866910976816709632
author Tian, Xinyu
Shen, Xiaotong
author_facet Tian, Xinyu
Shen, Xiaotong
contents This paper investigates the accuracy of generative models and the impact of knowledge transfer on their generation precision. Specifically, we examine a generative model for a target task, fine-tuned using a pre-trained model from a source task. Building on the "Shared Embedding" concept, which bridges the source and target tasks, we introduce a novel framework for transfer learning under distribution metrics such as the Kullback-Leibler divergence. This framework underscores the importance of leveraging inherent similarities between diverse tasks despite their distinct data distributions. Our theory suggests that the shared structures can augment the generation accuracy for a target task, reliant on the capability of a source model to identify shared structures and effective knowledge transfer from source to target learning. To demonstrate the practical utility of this framework, we explore the theoretical implications for two specific generative models: diffusion and normalizing flows. The results show enhanced performance in both models over their non-transfer counterparts, indicating advancements for diffusion models and providing fresh insights into normalizing flows in transfer and non-transfer settings. These results highlight the significant contribution of knowledge transfer in boosting the generation capabilities of these models.
format Preprint
id arxiv_https___arxiv_org_abs_2405_16837
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle Enhancing Accuracy in Generative Models via Knowledge Transfer
Tian, Xinyu
Shen, Xiaotong
Machine Learning
This paper investigates the accuracy of generative models and the impact of knowledge transfer on their generation precision. Specifically, we examine a generative model for a target task, fine-tuned using a pre-trained model from a source task. Building on the "Shared Embedding" concept, which bridges the source and target tasks, we introduce a novel framework for transfer learning under distribution metrics such as the Kullback-Leibler divergence. This framework underscores the importance of leveraging inherent similarities between diverse tasks despite their distinct data distributions. Our theory suggests that the shared structures can augment the generation accuracy for a target task, reliant on the capability of a source model to identify shared structures and effective knowledge transfer from source to target learning. To demonstrate the practical utility of this framework, we explore the theoretical implications for two specific generative models: diffusion and normalizing flows. The results show enhanced performance in both models over their non-transfer counterparts, indicating advancements for diffusion models and providing fresh insights into normalizing flows in transfer and non-transfer settings. These results highlight the significant contribution of knowledge transfer in boosting the generation capabilities of these models.
title Enhancing Accuracy in Generative Models via Knowledge Transfer
topic Machine Learning
url https://arxiv.org/abs/2405.16837