Neural Style Transfer with automatic style weight searching

Xiang Zhou
DOI: https://doi.org/10.54254/2755-2721/47/20241242
2024-03-15
Abstract:Neural Style Transfer (NST) has garnered significant attention in the field of computer vision. Previous research in this area has made important breakthroughs, creating various style transfer models and innovative architectural designs, and has achieved success in commercial applications. This paper presents a novel design for automatic style weight determination based on constraining the initial total loss within a healthy range and finding the optimal solution through grid search. This design aims to harness the potential of existing NST models by automating the optimization of neural style transfer hyperparameters. The paper first discusses the impact of content weights and loss weights on the generated images and validates the influence of weight ratios on image quality through experimental adjustments of weight proportions. It also confirms that the initial loss essentially defines the optimization space of the optimizer. The paper explores the significance of the initial loss and proposes a method to improve image generation quality by constraining the initial loss range.
What problem does this paper attempt to address?