Візуалізація результатів трансферного навчання на основі моделі MobileNet

Abstract

The paper analyzes the application of visualization methods for transfer learning results using the MobileNet model for media content classification and aggregation. Practical examples of using activation heatmaps (GradCAM) for explaining model decisions and identifying influential image regions are presented. The benefits of this approach for detecting overfitting issues and optimizing model parameters are explored.У роботі проаналізовано застосування методів візуалізації результатів трансферного навчання з використанням моделі MobileNet для класифікації та агрегування медіаконтенту. Наведено практичні приклади використання теплових карт активації (Grad-CAM) для пояснення рішень моделі та визначення дылянок зображень, що мають найбільший вплив на класифікацію. Досліджено переваги такого підходу для виявлення задач перенавчання та оптимізації параметрів моделі.