$$\rightleftharpoonup{xx}$$
$$\longleftharp{xx}$$,
$$\longrightharp{xx}$$,
Zaproponowana struktura SegCycle-SPADE została opracowana w celu rekonstrukcji i ulepszenia tradycyjnych wzorów dziedzictwa kulturowego poprzez segmentację semantyczną, wzmacnianie cech z wykorzystaniem mechanizmu uwagi, rekonstrukcję generatywną oraz syntezę stylu artystycznego. W badaniu wykorzystano ogólnodostępne zbiory danych obrazów artystycznych i dziedzictwa kulturowego, w tym zbiór danych Miao Batik oraz zbiór danych WikiArt. W badaniach nie uczestniczyli ludzie, nie wykorzystano danych pacjentów, informacji osobistych, zwierząt ani dokumentacji klinicznej; w związku z tym nie wymagano zgody instytucjonalnej komisji etycznej ani świadomej zgody uczestników. W pierwszej kolejności do ewaluacji wykorzystano zbiór motywów kulturowych Miao Batik oraz zbiór danych WikiArt. Zbiór danych Miao Batik został opisany w pracy Quan et al. (2024)32 i zawiera 15 148 adnotowanych obrazów tradycyjnych motywów batiku Miao. Istniejące adnotacje dostarczone przez twórców zbioru zostały użyte bezpośrednio; w niniejszym badaniu nie tworzono dodatkowych adnotacji ręcznych. Następnie przeprowadzono wstępne przetwarzanie obrazów poprzez zmianę rozmiaru, normalizację, odszumianie oraz tworzenie maski segmentacji. Dokładne parametry wstępnego przetwarzania opisano w podrozdziale Wstępne przetwarzanie danych. Zaproponowana struktura wykorzystuje model oparty na transformatorze o nazwie SegFormer-B2 do segmentacji semantycznej danych. Metoda ta została zaprojektowana w celu wykrywania kluczowych regionów motywów kulturowych i uczenia się ich struktur. Segmentowane cechy są następnie wzmacniane poprzez mechanizm uwagi, w którym podkreślane są istotne informacje związane z motywami kulturowymi, a informacje nieistotne są odrzucane. Konfiguracja mechanizmu uwagi została opisana w podrozdziale CAFFM. Wzmocnione cechy są następnie przekazywane do CycleGAN, gdzie uszkodzone struktury są rekonstruowane poprzez uczenie cykliczne. Podczas treningu sztucznie stworzono niekompletne próbki motywów, stosując operacje losowego maskowania i częściowego zasłaniania oryginalnych obrazów batiku Miao. Procedury degradacji te symulowały brakujące regiony motywów i uszkodzenia strukturalne powszechnie obserwowane w zdegradowanych artefaktach dziedzictwa kulturowego. Powstałe niekompletne obrazy posłużyły jako dane wejściowe dla modułu rekonstrukcji CycleGAN, natomiast odpowiadające im oryginalne obrazy stanowiły cele rekonstrukcji. Zrekonstruowane wzory dziedzictwa kulturowego są następnie ulepszane za pomocą SPADE, gdzie wzmocnienie artystyczne jest realizowane na podstawie wygenerowanych map segmentacji. Skuteczność proponowanej struktury oceniono za pomocą ilościowych metryk segmentacji i jakości obrazu, natomiast autentyczność wizualną zrekonstruowanych motywów kulturowych oceniono jakościowo. Autentyczność wizualną oceniano poprzez inspekcję wzrokową wygenerowanych wzorów kulturowych i nie była ona traktowana jako niezależna metryka ilościowa. Ocena koncentrowała się na zachowaniu motywów, spójności semantycznej, cechach kulturowych, kohezji strukturalnej oraz wyglądzie artystycznym w odniesieniu do odpowiadających im referencyjnych motywów kulturowych. Ilościowa ocena wydajności modelu została przeprowadzona z wykorzystaniem dokładności segmentacji (Segmentation Accuracy), IoU, współczynnika Dice (Dice Coefficient), wskaźnika podobieństwa strukturalnego (SSIM), szczytowego stosunku sygnału do szumu (PSNR) oraz odległości Frécheta Inception Distance (FID). Rysunek 2 przedstawia ogólny schemat działania proponowanej struktury SegCycle-SPADE i podsumowuje metryki oceny wykorzystane w niniejszym badaniu.

Rycina 2. Ogólny schemat przepływu architektury proponowanego frameworka SegCycle-SPADE. Przegląd kompletnego przepływu pracy SegCycle-SPADE. Obrazy ze zbiorów danych Miao Batik i WikiArt poddawane są wstępnemu przetwarzaniu przed etapem segmentacji semantycznej z wykorzystaniem SegFormer-B2, wzmocnienia cech za pomocą CAFFM, rekonstrukcji wzorów przy użyciu CycleGAN oraz generowania stylu artystycznego za pomocą SPADE. Wydajność modelu jest oceniana za pomocą wskaźników Segmentation Accuracy, Intersection over Union (IoU), współczynnika Dice, Structural Similarity Index Measure (SSIM), PSNR oraz Fréchet Inception Distance (FID), natomiast autentyczność wizualna jest oceniana jakościowo. Kliknij tutaj, aby wyświetlić powiększoną wersję tej ryciny.
Struktura SegCycle-SPADE do rekonstrukcji wzorów dziedzictwa kulturowego została zaprojektowana w celu zintegrowania wielu komponentów głębokiego uczenia (DL) dla dokładnej segmentacji motywów, rekonstrukcji i artystycznego ulepszenia, co zilustrowano na Rysunku 2. Do zapewnienia różnorodności wzorów kulturowych i stylów artystycznych wykorzystano obrazy z zestawu danych motywów kulturowych Miao Batik oraz zestawu danych WikiArt. Początkowo obrazy są przetwarzane za pomocą modułu segmentacji semantycznej opartego na modelu SegFormer, aby zidentyfikować i wyodrębnić motywy kulturowe z tła. Cała architektura składa się z czterech głównych etapów. Po pierwsze, model SegFormer wyodrębnia mapy segmentacji semantycznej z obrazów wzorów kulturowych. Po drugie, moduł CAFFM integruje cechy segmentacji z reprezentacjami generatywnymi w celu usprawnienia uczenia cech. Po trzecie, moduł CycleGAN rekonstruuje wzory kulturowe, wykorzystując scalone reprezentacje cech. Na koniec moduł SPADE generuje artystycznie ulepszone wyniki, zachowując spójność strukturalną dzięki syntezie sterowanej segmentacją. Uszczegółowione mapy cech są następnie przetwarzane przez moduł rekonstrukcji CycleGAN, który uczy się przywracać niekompletne motywy kulturowe poprzez mapowanie zdegradowanych wzorów na ich odpowiadające im pełne formy. Próbki niekompletnych motywów wygenerowano poprzez zastosowanie losowego maskowania i operacji częściowego zasłaniania oryginalnych obrazów Miao Batik, symulując w ten sposób brakujące regiony wzorów i degradację strukturalną często obserwowaną w uszkodzonych artefaktach kulturowych. Każdy zdegradowany obraz został sparowany z odpowiadającym mu obrazem oryginalnym, który służył jako cel rekonstrukcji podczas uczenia. Strategia ta umożliwiła modułowi CycleGAN naukę przywracania brakujących struktur motywów przy jednoczesnym zachowaniu spójności semantycznej i istotnych cech kulturowych. Ostatecznie generator SPADE tworzy wizualnie ulepszone wyniki artystyczne, warunkując syntezę obrazu na podstawie wygenerowanych map segmentacji, co pozwala zachować integralność strukturalną motywów kulturowych w całym procesie ulepszania artystycznego.
Poniższe kroki są zaangażowane w proponowany schemat SegCycle-SPADE.
Krok 1: Gromadzenie zbiorów danych
W celu realizacji celów nauki wzorców kulturowych oraz generowania stylów artystycznych wykorzystano dwa zbiory danych. Zbiór motywów kulturowych Miao Batik posłużył do dostarczenia informacji o tradycyjnych wzorach kulturowych. Zbiór ten jest publicznie dostępny za pośrednictwem Zenodo (https://doi.org/10.5281/zenodo.20807658) i zawiera 15 148 opisanych obrazów tradycyjnych motywów batikowych ludu Miao. Wykorzystano bezpośrednio istniejące adnotacje dostarczone przez twórców zbioru; w niniejszym badaniu nie tworzono dodatkowych adnotacji ręcznych. Zbiór danych WikiArt został wykorzystany do dostarczenia zróżnicowanych informacji o stylach artystycznych na potrzeby generowania stylu i został pobrany z publicznie dostępnego repozytorium WikiArt (https://www.wikiart.org/).
Krok 2: Przetwarzanie wstępne danych
Wszystkie obrazy poddano wstępnemu przetwarzaniu poprzez zmianę rozmiaru, normalizację oraz redukcję szumów. Wykorzystano istniejące adnotacje motywów kulturowych pochodzące z oryginalnych źródeł zbioru danych, aby wesprzeć etap segmentacji semantycznej. W ramach niniejszego badania nie przeprowadzono żadnych dodatkowych procedur adnotacji ani ponownego etykietowania.
Krok 3: Segmentacja wzorców semantycznych z wykorzystaniem SegFormer
Do segmentacji motywów kulturowych wykorzystano model SegFormer. Dokładna architektura szkieletowa (backbone) oraz strategia inicjalizacji modelu SegFormer zostały opisane w podrozdziale Semantic Pattern Segmentation Using SegFormer. W szczególności do semantycznej segmentacji motywów zastosowano architekturę SegFormer-B2 z szkieletem MIT-B2 wstępnie wytrenowanym na zbiorze ImageNet. Model ten skutecznie przechwytuje globalne informacje kontekstowe, zachowując jednocześnie złożone szczegóły strukturalne tradycyjnych wzorców kulturowych.
Krok 4: CAFFM
Moduł CAFFM łączy cechy segmentacji semantycznej z reprezentacjami generatywnymi, umożliwiając frameworkowi naukę zarówno charakterystyki motywów strukturalnych, jak i informacji o stylu artystycznym. Szczegóły integracji CAFFM przedstawiono w podrozdziale CAFFM. Moduł ten znajduje się pomiędzy etapem segmentacji semantycznej opartym na SegFormer a etapem rekonstrukcji generatywnej, gdzie scala cechy strukturalne pochodzące z segmentacji z cechami rekonstrukcji za pomocą wielogłowicowej uwagi krzyżowej (multi-head cross-attention). Proces ten poprawia zachowanie motywów kulturowych i zwiększa realizm zrekonstruowanych wyników.
Krok 5: Rekonstrukcja wzorów (CycleGAN)
Połączone cechy są następnie przetwarzane przez moduł CycleGAN w celu rekonstrukcji struktur wzorów kulturowych. Architektura CycleGAN oraz konfiguracja treningowa zostały opisane w podrozdziale Rekonstrukcja wzorów przy użyciu CycleGAN. Moduł rekonstrukcji składa się z dwóch generatorów i dwóch dyskryminatorów, wykorzystuje architekturę generatora opartą na sieciach rezdualnych z dziewięcioma blokami rezdualnymi, stosuje dyskryminator PatchGAN i jest trenowany z wykorzystaniem strat adversarialnych oraz strat spójności cyklu. Taka konfiguracja umożliwia dwukierunkowe mapowanie domen i ułatwia rekonstrukcję niekompletnych struktur wzorów kulturowych.
Krok 6: Generowanie stylu artystycznego (SPADE)
Zrekonstruowane wzory są następnie przetwarzane przez moduł SPADE w celu przeprowadzenia syntezy stylu artystycznego sterowanej segmentacją. Konfiguracja generatora SPADE została opisana w podrozdziale Artistic Style Generation Using SPADE. Moduł wykorzystuje bloki rezidualne SPADE, warstwy normalizacji adaptacyjnej przestrzennie oraz warunkowanie semantyczne pochodzące z map segmentacji SegFormer i reprezentacji cech CAFFM. Dzięki warunkowaniu generowania obrazów na podstawie map segmentacji, zsyntetyzowane wyniki zachowują spójność strukturalną z oryginalnymi motywami kulturowymi, jednocześnie przyjmując charakterystyczne cechy stylu artystycznego.
Krok 7: Ocena wydajności
Zaproponowany model został oceniony przy użyciu wielu metryk segmentacji i oceny jakości obrazu, w tym dokładności segmentacji (Segmentation Accuracy), IoU, współczynnika podobieństwa Dice’a (Dice), SSIM, PSNR oraz FID. Aby ocenić spójność eksperymentalną, wszystkie doświadczenia powtórzono pięć razy, stosując różne ziarna losowe, a wyniki przedstawiono jako średnią ± odchylenie standardowe. Istotność statystyczną oceniono za pomocą testów t-studenta dla prób zależnych, przyjmując próg istotności p < 0,05.
Zbieranie zbiorów danych
W proponowanej strukturze SegCycle-SPADE do rozumienia wzorców kulturowych i nauki stylu wykorzystuje się dwa zbiory danych. Pierwszym zbiorem danych zastosowanym w proponowanej strukturze jest zbiór motywów kulturowych batiku ludu Miao. Zbiór ten zawiera motywy kulturowe batiku Miao, które są zazwyczaj obrazami tradycyjnego batiku Miao zawierającymi motywy takie jak zwierzęta, rośliny i kształty geometryczne, stosowane w sztuce etnicznej ludu Miao. Zbiór ten zawiera obrazy z bogatymi informacjami o teksturze, co jest zazwyczaj istotne dla zachowania dziedzictwa kulturowego. Drugim zbiorem danych wykorzystanym w proponowanej strukturze SegCycle-SPADE jest zbiór danych WikiArt. Zbiór ten zawiera ogromną liczbę dzieł sztuki, zazwyczaj w różnych stylach. Jest on wykorzystywany do złożonej nauki stylu, co jest zazwyczaj przydatne w celu ulepszania dzieł sztuki. Zbiór ten obejmuje 80 000 dzieł w rozdzielczości HD, z 27 różnymi wzorami, co czyni go bardziej użytecznym w zadaniach generowania lub transformacji stylu opartych na DL.
Zbiór danych Miao Batik:
Zbiór danych Miao Batik (https://doi.org/10.5281/zenodo.20807658) obejmuje 15 148 obrazów wzorów batiku przedstawiających motywy o znaczeniu kulturowym. Obrazy zawierają różnorodne tradycyjne wzory, takie jak motywy zwierzęce (np. motyle i ryby), wzory roślinne oraz motywy geometryczne niosące symbolikę kulturową. Zbiór ten jest istotnym elementem proponowanego modelu, ponieważ dostarcza autentycznych struktur kulturowych, które umożliwiają modułowi segmentacji dokładne identyfikowanie i zachowywanie granic motywów podczas rekonstrukcji wzoru (Tabela 1). W niniejszym badaniu motywy o znaczeniu kulturowym odnoszą się do symbolicznych elementów wizualnych powszechnie dokumentowanych w literaturze dotyczącej dziedzictwa kulturowego ludu Miao i reprezentowanych w adnotacjach zbioru danych, w tym ptaków, motyli, wzorów kwiatowych oraz totemów przodków. Motywy te wybrano na podstawie ich udokumentowanego znaczenia kulturowego i powtarzalnej obecności w tradycyjnych wzorach batiku i haftów Miao, a nie wyłącznie na podstawie ich częstotliwości występowania czy kompozycji przestrzennej. Adnotacje motywów i etykiety segmentacji opracowane pod kierunkiem ekspertów pobrano z oryginalnego źródła zbioru danych Miao Batik i wykorzystano bezpośrednio w niniejszym badaniu. Autorzy nie przeprowadzali żadnych dodatkowych adnotacji ręcznych, ponownego etykietowania ani procedur adnotacji z udziałem ekspertów. Istniejące adnotacje dostarczone przez twórców zbioru danych wykorzystano do trenowania i oceny segmentacji semantycznej.
| Parametr | Opis |
| Nazwa zbioru danych | Zbiór danych wzorów kulturowych batiku ludu Miao |
| Źródło zbioru danych | Repozytorium Zenodo (DOI: 10.5281/zenodo.20807658) |
| Dziedzina | Niematerialne dziedzictwo kulturowe (ICH) / analiza wzorów tekstylnych |
| Całkowita liczba obrazów | 15 148 obrazów |
| Typ obrazu | Cyfrowe obrazy tradycyjnych tkanin batiki ludu Miao |
| Kategorie wzorów | Motywy zwierzęce, roślinne i geometryczne |
| Pochodzenie kulturowe | Kultura etniczna Miao, prowincja Guizhou, Chiny |
| Oryginalna rozdzielczość obrazu | Obrazy o wysokiej rozdzielczości (zazwyczaj ≥ 1 000 pikseli na krótszym boku), pozyskane jako skany surowe lub fotografie przed wstępnym przetwarzaniem |
| Rozdzielczość treningowa | Obrazy przeskalowane do rozmiaru 256 × 256 pikseli podczas wstępnego przetwarzania |
| Dostępność adnotacji | Istniejące adnotacje segmentacyjne dostarczone przez twórców zbioru danych zostały wykorzystane bez modyfikacji do trenowania i ewaluacji. W niniejszym badaniu nie przeprowadzono dodatkowych adnotacji ręcznych, ponownego etykietowania ani procedur potwierdzania przez ekspertów. |
| Zastosowanie w niniejszym badaniu | Segmentacja motywów kulturowych, ekstrakcja cech, zachowanie motywów i rekonstrukcja wzorów |
Tabela 1: Charakterystyka zbioru danych wzorów kulturowych batików ludu Miao. Podsumowanie zbioru danych motywów kulturowych batików ludu Miao wykorzystanego do segmentacji semantycznej, analizy wzorów kulturowych oraz rekonstrukcji motywów. Tabela opisuje źródło zbioru danych, charakterystykę obrazów, kategorie motywów, pochodzenie kulturowe, rozdzielczość obrazów oraz rolę zbioru w ramach zaproponowanej architektury SegCycle-SPADE.
Zbiór danych WikiArt:
Zbiór danych WikiArt [https://www.wikiart.org/] jest popularnym, wielkoskalowym cyfrowym repozytorium sztuki, które obejmuje ponad 80 000 obrazów reprezentujących 27 różnych stylów artystycznych wymienionych w Tabeli 2. Style te obejmują sztukę renesansu, impresjonizm, kubizm i surrealizm. Zbiór ten jest bardzo istotny w proponowanym modelu, ponieważ dostarcza wiedzy pozwalającej modułowi SPADE na tworzenie wzorów wzbogaconych kulturowo przy jednoczesnym zachowaniu informacji strukturalnych uzyskanych w procesie segmentacji. Zbiór danych składa się z 56 000 obrazów do uczenia oraz 12 000 obrazów do walidacji i testowania. Obrazy zawarte w zbiorze pomagają w efektywnym trenowaniu modelu DL.
| Parametr | Opis |
| Nazwa zbioru danych | Zbiór danych WikiArt |
| Źródło zbioru danych | Repozytorium WikiArt (https://www.wikiart.org/) |
| Dziedzina | Sztuka cyfrowa i malarstwo |
| Całkowita liczba obrazów | Około 80 000 dzieł sztuki |
| Obrazy treningowe | 56 000 |
| Obrazy walidacyjne | 12 000 |
| Obrazy testowe | 12 000 |
| Style artystyczne | 27 stylów artystycznych, w tym renesans, impresjonizm, kubizm, surrealizm, ekspresjonizm, realizm i sztuka abstrakcyjna |
| Oryginalna rozdzielczość obrazów | Oryginalne rozdzielczości obrazów różnią się w zależności od dzieła i źródła. Podczas wstępnego przetwarzania obrazy zostały przeskalowane do jednolitej rozdzielczości 256 × 256 pikseli. |
| Rozdzielczość treningowa | Obrazy przeskalowane do 256 × 256 pikseli podczas wstępnego przetwarzania przed nauką stylu artystycznego i syntezą obrazu sterowaną segmentacją. |
| Podział zbioru danych | Stratyfikowany podział losowy (70% trening, 15% walidacja i 15% testy) przy użyciu stałego ziarna losowania 42 |
| Strategia próbkowania stylów | Zastosowano stratyfikowane próbkowanie proporcjonalne, aby zachować rozkład 27 stylów artystycznych w podzbiorach treningowym, walidacyjnym i testowym |
| Zastosowanie w niniejszym badaniu | Nauka stylu artystycznego, reprezentacja stylu oraz sterowana segmentacją synteza artystyczna |
Tabela 2: Charakterystyka zbioru danych WikiArt.Podsumowanie zbioru danych WikiArt wykorzystanego do uczenia stylu artystycznego i syntezy obrazów sterowanej stylem. Tabela opisuje źródło zbioru danych, rozkład obrazów, zakres stylów artystycznych, podział zbioru danych, rozdzielczość obrazów oraz zastosowanie zbioru w ramach proponowanego frameworku SegCycle-SPADE.
Zbiór danych Miao Batik zawiera 15 148 obrazów reprezentujących tradycyjne motywy kulturowe ludu Miao.32 Zbiór ten jest ogólnodostępny za pośrednictwem Zenodo (https://doi.org/10.5281/zenodo.20807658). Przed trenowaniem modelu wszystkie obrazy poddano inspekcji wizualnej, zmieniono ich rozmiar do 256 × 256 pikseli, znormalizowano oraz przeanalizowano pod kątem uszkodzonych lub zduplikowanych próbek. Kontrola jakości nie doprowadziła do wykluczenia żadnych obrazów, w związku z czym wszystkie 15 148 obrazów zachowano do dalszej analizy. Zbiór danych został losowo podzielony na podzbiory treningowy (70%), walidacyjny (15%) i testowy (15%) przy użyciu stałego ziarna losowości (random seed) o wartości 42, aby zapewnić powtarzalność podziału danych. Dostęp do zbioru danych WikiArt uzyskano poprzez oficjalne repozytorium WikiArt (https://www.wikiart.org/), które zawiera ponad 80 000 dzieł sztuki obejmujących 27 stylów artystycznych. W eksperymentach dotyczących uczenia stylu wykorzystano 56 000 obrazów do trenowania, 12 000 do walidacji oraz 12 000 do testowania.
Wstępne przetwarzanie danych
Przed rozpoczęciem trenowania proponowanego frameworka SegCycle-SPADE, zbiór motywów kulturowych Miao Batik oraz zbiór WikiArt zostały poddane kilku etapom wstępnego przetwarzania, aby zapewnić spójną jakość obrazów i dokładną reprezentację cech. Do obu zbiorów danych zastosowano ten sam podstawowy potok przetwarzania, obejmujący odszumianie Gaussa, normalizację, zmianę rozmiaru do spójnej rozdzielczości wejściowej oraz weryfikację jakości obrazu. Zbiory te pełniły jednak odmienne role w ramach frameworka. Zbiór WikiArt wykorzystano do nauki i syntezy stylu artystycznego, natomiast zbiór Miao Batik służył głównie do segmentacji i rekonstrukcji motywów kulturowych. Poza tymi specyficznymi dla zadań rolami nie wprowadzono żadnych zmian w przetwarzaniu danych. Aby zapewnić spójne przetwarzanie przez model DL, obrazy zostały najpierw przeskalowane do stałej rozdzielczości. Krok ten był niezbędny, ponieważ obrazy w obu zbiorach różniły się rozdzielczością w zależności od źródła. Zmiana rozmiaru zwiększyła wydajność obliczeniową i zapobiegła wpływowi niespójności wymiarowych na proces trenowania. Drugim etapem wstępnego przetwarzania była normalizacja, w której wartości pikseli przeskalowano do standardowego zakresu, aby ustabilizować aktualizacje gradientów podczas trenowania. Następnie obrazy przetworzono za pomocą filtrowania Gaussa w celu zredukowania szumów, które mogłyby zakłócać struktury motywów kulturowych. W przypadku zbioru Miao Batik, istniejące maski segmentacji motywów kulturowych, pozyskane bezpośrednio z oryginalnego źródła zbioru danych, zostały wykorzystane bez modyfikacji do trenowania i ewaluacji segmentacji semantycznej. Na potrzeby niniejszego badania nie generowano nowych masek segmentacji.
O ile nie wskazano inaczej, równania opisujące uznane metody zostały zaadaptowane z wcześniejszych badań i są odpowiednio cytowane. Równania opisujące proponowaną metodę CAFFM oraz złożony framework optymalizacyjny SegCycle-SPADE zostały opracowane przez autorów na potrzeby niniejszej pracy.
Przyjmijmy, że obraz wejściowy z zestawu danych jest reprezentowany przez Równanie 1:
(1)
W tym przypadku H, W i C oznaczają odpowiednio wysokość obrazu, szerokość oraz liczbę kanałów kolorystycznych. Wszystkie obrazy są zmieniane na stały wymiar H′ × W′, zgodnie z Równaniem 2:

W tym przypadku Ir oznacza obraz po zmianie rozmiaru. Znormalizowane wartości pikseli są obliczane zgodnie z Równaniem 3:
(3)
Pomaga to ustabilizować procedurę uczenia. Filtrowanie szumów jest wykonywane przy użyciu filtru Gaussa, zgodnie z Równaniem 4:

Tutaj G(σ) oznacza filtr Gaussa, a * reprezentuje splot. Zastosowano filtr Gaussa z jądrem o rozmiarze 5 × 5 i odchyleniem standardowym σ = 1.0. W celu zredukowania artefaktów krawędziowych zastosowano do pikseli brzegowych dopełnienie zwierciadlane (reflective padding). Proces odszumiania przeprowadzono bezpośrednio po wczytaniu obrazów, przed skalowaniem i normalizacją. Aby zapewnić jednolity preprocessing w całym badaniu, tę samą konfigurację filtra zastosowano do każdego obrazu w zbiorach danych Miao Batik i WikiArt. Dla zbioru danych Miao Batik maski segmentacji tworzy się zgodnie z Równaniem 5:

W tym przypadku M jest maską segmentacji służącą do naprowadzania modelu SegFormer.
Potok przetwarzania wstępnego rozpoczyna się od pozyskania obrazów z zestawu danych Miao Batik oraz zestawu danych WikiArt, jak pokazano na Rysunku 3. Podczas trenowania nie zastosowano technik augmentacji danych. Po przetwarzaniu wstępnym, które obejmowało zmianę rozmiaru, normalizację, odszumianie gaussowskie oraz przygotowanie masek segmentacyjnych, obrazy zostały bezpośrednio wykorzystane do trenowania, walidacji i testowania proponowanego frameworku SegCycle-SPADE. Pierwszym krokiem potoku przetwarzania wstępnego jest zmiana rozmiaru obrazów do stałej wartości, co pozwala modelowi DL na bardziej efektywne przetwarzanie danych. Drugim krokiem jest normalizacja, która przekształca wartości pikseli na ustandaryzowany zakres numeryczny i ułatwia konwergencję modelu. Trzeci krok obejmuje usuwanie szumów, w wyniku czego obrazy są oczyszczane z niepożądanych zakłóceń w celu poprawy rozpoznawania wzorców. Dodatkowo, w przypadku zestawu danych Miao Batik, adnotowane są obszary motywów kulturowych, co umożliwia wygenerowanie masek wykorzystywanych w module segmentacji proponowanego modelu, zapewniając zachowanie motywów kulturowych podczas generowania. Końcowym wynikiem tego etapu jest oczyszczony zestaw danych, gotowy do trenowania modułów segmentacji i generowania proponowanego modelu.

Rycina 3. Potok wstępnego przetwarzania danych dla obrazów dziedzictwa kulturowego. Przebieg procesu ilustrujący procedury wstępnego przetwarzania obrazów stosowane przed trenowaniem modelu. Potok obejmuje pozyskiwanie zbioru danych, zmianę rozmiaru obrazów, normalizację, odszumianie gaussowskie, adnotacje istniejących motywów kulturowych oraz przygotowanie masek segmentacji. Powstałe przetworzone obrazy i maski służą jako dane wejściowe dla segmentacji semantycznej i rekonstrukcji wzorców. Kliknij tutaj, aby wyświetlić powiększoną wersję tej ryciny.
Każdy obraz został poddany procesowi kontroli jakości przed trenowaniem modelu. Zbiór danych Miao Batik zawierał 15 148 obrazów. Próbki uszkodzone, zdublowane oraz niskiej jakości zostały już opracowane przez oryginalnych twórców zbioru danych; w związku z tym w ramach niniejszego badania podczas przesiewowej kontroli jakości nie wykluczono żadnych dodatkowych obrazów. W konsekwencji do analizy zachowano wszystkie 15 148 obrazów. Podczas wstępnego przetwarzania obrazy wczytywano w formacie RGB i zmieniano ich rozmiar do 256 × 256 pikseli przy użyciu interpolacji bilinearnej. Wartości pikseli przeskalowano z zakresu [0, 255] do [0, 1] poprzez dzielenie przez 255. Następnie zastosowano normalizację dla poszczególnych kanałów, używając wartości średnich [0.485, 0.456, 0.406] oraz odchyleń standardowych [0.229, 0.224, 0.225] odpowiednio dla kanałów czerwonego, zielonego i niebieskiego. Potok wstępnego przetwarzania obejmował wczytywanie obrazów, konwersję do RGB, zmianę rozmiaru, skalowanie wartości pikseli, normalizację oraz konwersję do tensorów. W razie potrzeby obrazy w skali szarości konwertowano do trójkanałowego formatu RGB, aby zachować kompatybilność z modelami DL. Po wstępnym przetwarzaniu obrazy podzielono na podzbiory treningowy, walidacyjny i testowy. Wszystkie etapy frameworka SegCycle-SPADE – w tym segmentacja semantyczna, fuzja cech, rekonstrukcja motywu oraz synteza artystyczna oparta na SPADE – wykorzystywały spójną rozdzielczość wejściową 256 × 256 pikseli.
Segmentacja wzorców semantycznych przy użyciu SegFormer
Po tym etapie wstępnego przetwarzania, oczyszczone i znormalizowane obrazy z zestawu danych motywów kulturowych Miao Batik oraz zestawu danych WikiArt są wprowadzane do modułu segmentacji semantycznej opartego na zaproponowanej strukturze SegFormer-B2. Celem tego kroku jest prawidłowe zidentyfikowanie i oddzielenie motywów kulturowych występujących we wzorach dziedzictwa. Zaproponowana struktura SegFormer opiera się na architekturze transformera, która zawiera hierarchiczny koder Transformer oraz lekki dekoder MLP, aby precyzyjnie przechwytywać globalne informacje kontekstowe, a także szczegółowe informacje strukturalne ze złożonych wzorów dziedzictwa. Model najpierw wyodrębnia reprezentacje cech w wielu skalach z obrazu wejściowego, a następnie dokonuje fuzji tych reprezentacji poprzez dekoder w celu wygenerowania dokładnych segmentowanych wzorców. Zapewnia to prawidłową segmentację wzorów na obrazach dziedzictwa na takie motywy jak wzory geometryczne, roślinne i symboliczne, oddzielając je tym samym od tła przy zachowaniu ich integralności strukturalnej. Informacje strukturalne te są następnie wykorzystywane w późniejszych etapach generowania wzorców w ramach struktury SegCycle-SPADE.
Niech wstępnie przetworzony obraz wejściowy będzie reprezentowany przez Równanie 6:
(6)
Koder SegFormer dzieli obraz na fragmenty (patche) i ekstrahuje cechy hierarchiczne przy użyciu warstw transformatora, jak pokazano w Równaniu 71:

Tutaj F oznacza wielkoskalowe mapy cech uzyskane z koderen transformera. Cechy te kodują zarówno lokalne wzorce tekstury, jak i globalne relacje kontekstowe pomiędzy regionami motywów. Model SegFormer wyodrębnia mapy cech w różnych skalach, zgodnie z definicją w Równaniu 8:

Zastosowanie reprezentacji wieloskalowych ułatwia wykrywanie wzorców o różnych rozmiarach w obrębie motywów kulturowych. Ostatecznie cechy zostają połączone przy użyciu dekodera MLP, zgodnie z równaniem 91:

W tym przypadku S oznacza końcową przewidywaną mapę segmentacji.
Szkielet SegFormer-B2 został zainicjowany przy użyciu wag wstępnie wytrenowanych na zbiorze ImageNet, a następnie dostrojony (fine-tuned) na zbiorze danych Miao Batik w celu segmentacji motywów kulturowych. Punkt kontrolny (checkpoint) z wstępnym treningiem pobrano z oficjalnej implementacji SegFormer. W szczególności, do inicjalizacji szkieletu SegFormer-B2 przed dostrojeniem wykorzystano punkt kontrolny MIT-B2 (mit_b2.pth) wstępnie wytrenowany na zbiorze ImageNet-1K, udostępniony w oficjalnym repozytorium SegFormer. Wagi te odpowiadają szkieletowi MIT-B2 udostępnionemu przez autorów SegFormer i służyły jako model inicjalizujący dla treningu segmentacji semantycznej. Pozostałe warstwy specyficzne dla zadania zostały zainicjowane przy użyciu domyślnych procedur inicjalizacji wag w PyTorch przed rozpoczęciem treningu.
W architekturze zastosowano czterostopniowy hierarchiczny koder Transformer z nakładającymi się osadzeniami łatek (patch embeddings). W etapie początkowym rozmiar łatki ustawiono na 7 × 7 z krokiem 4. Dla każdego z czterech etapów kodera wymiary osadzeń wynosiły odpowiednio 64, 128, 320 i 512. W czterech etapach zastosowano kolejno 1, 2, 5 i 8 głowic wielogłowicowej mechanizmu self-attention, a odpowiadające im głębokości kodera wynosiły 3, 4, 6 i 3 warstwy. Cechy wieloskalowe pobrane z kodera zostały zagregowane za pomocą lekkiego dekodera opartego wyłącznie na MLP. Przed utworzeniem końcowej mapy segmentacji dekoder rzutował cechy z każdego etapu kodera do jednej wspólnej przestrzeni osadzeń. We wszystkich blokach Transformera wykorzystano funkcję aktywacji Gaussian Error Linear Unit (GELU). W celu poprawy generalizacji i ograniczenia przeuczenia podczas treningu zastosowano współczynnik dropout wynoszący 0,1.
Podczas fazy szkolenia framework SegFormer otrzymuje wstępnie przetworzone obrazy z obu zbiorów danych. Obrazy z zestawu danych Miao Batik zawierają obszary motywów, które służą jako etykiety do nadzorowanego uczenia modelu segmentacji. Koder Transformer przetwarza cały obraz i przechwytuje zależności dalekiego zasięgu między komponentami obrazu, co jest szczególnie istotne w przypadku tradycyjnych wzorów batik zawierających rozproszone przestrzennie motywy. Hierarchiczny mechanizm ekstrakcji cech jednocześnie przechwytuje struktury lokalne, takie jak powtarzalne tekstury geometryczne. Dekoder MLP przetwarza te wyekstrahowane cechy i generuje maskę segmentacji wyróżniającą obszary motywów. Mapy segmentacji wygenerowane na tym etapie są następnie wykorzystywane jako wejścia strukturalne dla modułu uwagi oraz etapu rekonstrukcji wzoru, co pomaga zachować autentyczność generowanych wzorów.
Motywy kulturowe podzielono na różne klasy do segmentacji semantycznej zgodnie z ich cechami wizualnymi i kulturowymi. Taksonomia segmentacji obejmowała motywy zwierzęce (np. motyle, ryby, ptaki i inną symboliczną faunę), motywy roślinne (np. kwiaty, liście, pnącza i wzory botaniczne), motywy geometryczne (np. powtarzające się kształty, obramowania i abstrakcyjne struktury geometryczne) oraz obszary tła reprezentujące strefy pozbawione motywów. Maski segmentacji na poziomie pikseli zostały wykorzystane do przypisania każdego piksela do jednej z zdefiniowanych wcześniej klas. Etykiety całkowite zakodowano w następujący sposób: Etykieta 0 = tło, Etykieta 1 = motywy zwierzęce, Etykieta 2 = motywy roślinne i Etykieta 3 = motywy geometryczne. Adnotacje segmentacji i etykiety motywów pozyskano bezpośrednio z oryginalnego źródła zbioru danych Miao Batik. W niniejszym badaniu nie przeprowadzono dodatkowej adnotacji ręcznej, zmiana etykietowania, weryfikacji granic ani procedur potwierdzenia przez ekspertów. Istniejące adnotacje dostarczone przez twórców zbioru danych zostały wykorzystane bez modyfikacji do celów uczenia i ewaluacji.
Model SegFormer do segmentacji motywów kulturowych przetwarza wstępnie przetworzone obrazy z zestawu danych Miao Batik oraz zestawu danych WikiArt, wykorzystując mechanizm oparty na transformerze do dokładnego wykrywania obszarów wzorów kulturowych, jak pokazano na Rysunku 4. Najpierw do modelu SegFormer przekazywany jest obraz wejściowy zawierający tradycyjne motywy kulturowe. Koder Transformera wyodrębnia zarówno globalne informacje kontekstowe, jak i lokalne cechy strukturalne z fragmentów obrazu. Uzyskane cechy wieloskalowe są przekazywane do dekodera segmentacji, który wykorzystuje sieć Mix Feed-Forward Network w celu doprecyzowania reprezentacji cech i poprawy wykrywania motywów. Wynikiem modelu SegFormer jest maska segmentacyjna, która wyróżnia piksele odpowiadające wzorom kulturowym, jednocześnie tłumiąc nieistotne informacje tła. Wyizolowane wzory kulturowe służą następnie jako wytyczne strukturalne dla kolejnych etapów frameworku SegCycle-SPADE.

Rysunek 4. Segmentacja semantyczna motywów kulturowych w oparciu o SegFormer-B2. Architektura modułu segmentacji semantycznej SegFormer-B2 wykorzystanego do ekstrakcji motywów kulturowych i wytrenowanego wyłącznie na zbiorze danych Miao Batik. Struktura składa się z kodera opartego na Transformerze do wieloskalowej ekstrakcji cech oraz lekkiego dekodera segmentacji do generowania masek motywów. Model przewiduje cztery klasy semantyczne — zwierzęcia, rośliny, geometrię i tło — gdzie wynikowe maski segmentacji identyfikują istotne kulturowo obszary motywów, jednocześnie tłumiąc nieistotne informacje o tle. Wyniki tej segmentacji stanowią wytyczne strukturalne dla kolejnych etapów fuzji cech, rekonstrukcji i syntezy artystycznej w proponowanej strukturze SegCycle-SPADE. Prosimy kliknąć tutaj, aby wyświetlić powiększoną wersję tego rysunku.
W sugerowanych ramach nie ma potrzeby tworzenia nowych adnotacji segmentacji. Zbiór danych Miao Batik został pozyskany z istniejącego publicznego źródła, a model został przeszkolony przy użyciu powiązanych informacji o motywach dostarczonych przez twórców zbioru. W procesie uczenia model SegFormer generował mapy segmentacji semantycznej. W konsekwencji obecne badanie nie wymagało zastosowania dodatkowego oprogramowania do adnotacji ręcznych, wytycznych dotyczących adnotacji ani procedur kontroli jakości adnotacji.
CAFFM
Aby poprawić interakcję między cechami segmentacji semantycznej a reprezentacjami rekonstrukcji generatywnej, w badaniu zaproponowano również moduł CAFFM. Koder SegFormer-B2 dostarcza mapy cech semantycznych do modułu CAFFM, natomiast etap rekonstrukcji CycleGAN dostarcza mapy cech generatywnych. Najpierw wykorzystano warstwy projekcji liniowej do rzutowania obu strumieni cech do wspólnej przestrzeni osadzeń. Do obliczeń uwagi krzyżowej (cross-attention) stworzono reprezentacje zapytania (Q), klucza (K) i wartości (V) przy użyciu wygenerowanych tensorów cech. Następnie relacje między informacjami o motywach strukturalnych a elementami stylu artystycznego zostały zamodelowane za pomocą wielogłowicowej uwagi krzyżowej. Do połączonych reprezentacji cech zastosowano następnie normalizację warstwową, połączenia rezydualne oraz warstwy przejścia w przód z aktywacją GELU. Etap syntezy oparty na SPADE otrzymuje wynik z modułu CAFFM, co pozwala na zachowanie motywów o znaczeniu kulturowym bez utraty spójności artystycznej.
Aby zagwarantować kompatybilność cech, cechy wejściowe są najpierw rzutowane za pomocą warstw projekcji liniowej na wspólną przestrzeń zanurzeń o wymiarze 256. Interakcje między semantycznymi informacjami strukturalnymi a generatywnymi reprezentacjami stylu są następnie modelowane przy użyciu mechanizmu MultiHead Cross-Attention z ośmioma głowicami uwagi. Obliczenia uwagi krzyżowej (cross-attention) wykorzystują wektory Query (Q), Key (K) oraz Value (V), które są generowane przez specyficzne warstwy transformacji liniowej. Aby zwiększyć stabilność treningu i zachować integralność cech, zastosowano połączenia rezydualne oraz normalizację warstwową (Layer Normalization), co pozwala na dalszą poprawę reprezentacji cech po fuzji. Nieliniowe uczenie cech jest następnie usprawniane poprzez zastosowanie sieci typu feed-forward z funkcją aktywacji Gaussian Error Linear Unit (GELU). Moduł generuje reprezentację cechy wyjściowej o wymiarze 256, która jest przesyłana do kolejnych etapów syntezy kreatywnej. Moduł CAFFM skutecznie łączy dane dotyczące stylu artystycznego ze strukturami motywów kulturowych, wykorzystując technikę fuzji opartą na uwadze krzyżowej, co poprawia zachowanie motywów i spójność wizualną w zrekonstruowanych wzorach dziedzictwa kulturowego.
W proponowanym systemie cechy strukturalne pochodzą ze zbioru danych Miao Batik, natomiast cechy stylistyczne są wyuczane ze zbioru danych WikiArt. Przyjmijmy, że mapa cech pochodząca z sieci segmentacyjnej SegFormer z wykorzystaniem obrazów ze zbioru danych Miao Batik jest oznaczona jako Fs, natomiast mapa cech pochodząca z gałęzi ekstrakcji cech stylistycznych z wykorzystaniem obrazów WikiArt jest oznaczona jako Fa. Proponowany moduł CAFFM łączy obie dziedziny cech za pomocą mechanizmu wzajemnej uwagi (cross-attention), aby wyuczyć zależności strukturalnych i stylistycznych wzorów kulturowych. Tabela 3 przedstawia wszystkie charakterystyki architektury, w tym wymiary osadzeń, warstwy normalizacji, funkcje aktywacji oraz konfigurację głowic uwagi. Dla rozmiaru obrazu wejściowego 256 × 256 pikseli koder SegFormer-B2 generował semantyczne mapy cech o rozmiarze 64 × 64 × 128, podczas gdy gałąź ekstrakcji cech stylistycznych generowała mapy cech o rozmiarze 64 × 64 × 128. Obie mapy cech zostały rzutowane przez wyuczalne warstwy liniowe do wspólnej przestrzeni osadzeń o wymiarze 256 przed fuzją za pomocą mechanizmu wzajemnej uwagi.
| Parametr | Konfiguracja |
| Proponowany schemat | SegCycle-SPADE |
| Model segmentacji semantycznej | SegFormer-B2 |
| Etapy kodera SegFormer | 4 |
| Inicjalizacja wag | SegFormer-B2 wstępnie wytrenowany na zbiorze ImageNet-1K (szkielet MIT-B2) |
| Źródło punktu kontrolnego | Oficjalne repozytorium NVIDIA SegFormer (https://github.com/NVlabs/SegFormer); punkt kontrolny: segformer.b2.512x512.ade.160k |
| Strategia dostrajania | Kompletne dostrajanie (end-to-end fine-tuning) na zbiorze danych Miao Batik |
| Rozmiar osadzenia patcha | 7 × 7 |
| Krok patcha | 4 |
| Wymiary zanurzeń | 64, 128, 320 i 512 |
| Głębokości enkoderów | 3, 4, 6 i 3 |
| Głowice uwagi | 1, 2, 5 i 8 |
| Typ dekodera | Dekoder w całości oparty na MLP |
| Funkcja aktywacji | GELU |
| Wskaźnik rezygnacji | 0.1 |
| Moduł wzmacniania cech | Moduł fuzji cech kulturowych z mechanizmem uwagi krzyżowej (CAFFM) |
| Wymiar zanurzenia CAFFM | 256 |
| Głowice uwagi CAFFM | 8 |
| Skale fuzji CAFFM | 4 |
| Model rekonstrukcji | CycleGAN |
| Model generowania stylu | SPADE |
| Zbiór danych kulturowych | Zbiór danych Miao Batik |
| Zbiór danych stylów | Zbiór danych WikiArt |
| Obrazy batikowe ludu Miao | 15,148 |
| Obrazy WikiArt | Około 80 000 |
| Rozdzielczość obrazu wejściowego | 256 × 256 pikseli |
| Format kolorów | RGB |
| Metoda interpolacji | Interpolacja biliniowa |
| Metoda odszumiania | Filtrowanie gaussowskie |
| Rozmiar jądra Gaussa | 5 × 5 |
| Sigma Gaussa (σ) | 1 |
| Zakres normalizacji | [0, 1] |
| Wielkość partii | 16 |
| Liczba epok | 100 |
| Optymalizator | Adam |
| Szybkość uczenia | 0.0002 |
| Parametry optymalizatora Adam | β₁ = 0,5, β₂ = 0,999, ε = 1 × 10⁻⁸, weight decay = 0 |
| Funkcje straty | strata segmentacji, strata adversariana, strata spójności cyklu, strata rekonstrukcji, strata zachowania motywu oraz strata spójności stylu |
| Strategia podziału zbioru danych | Trening / Walidacja / Testowanie |
| Zbiór treningowy | 70% |
| Zbiór walidacyjny | 15% |
| Zbiór testowy | 15% |
| Ziarno losowe | 42 |
| Metryki oceny | Dokładność segmentacji, IoU, współczynnik Dice'a, SSIM, PSNR i FID |
| Język programowania | Python 3.8.10 |
| Platforma głębokiego uczenia | PyTorch 1.10.0 |
| Platforma sprzętowa | Karta GPU NVIDIA RTX 3090 (24 GB VRAM), procesor Intel Core i7 (11. generacji), 32 GB RAM |
| Środowisko operacyjne | Ubuntu 20.04 LTS |
Tabela 3: Układ eksperymentalny i konfiguracja modelu. Podsumowanie komponentów architektonicznych, konfiguracji treningu, ustawień wstępnego przetwarzania, zbiorów danych, parametrów optymalizacji, metryk ewaluacji oraz środowiska obliczeniowego wykorzystanego do implementacji i oceny proponowanego frameworka SegCycle-SPADE.
Moduł uwagi najpierw rzutuje mapę cech na reprezentacje zapytania (query), klucza (key) i wartości (value) zgodnie z Równaniem 10:

Tutaj Wq, Wk i Wv są uczącymi się macierzami wag. Wagi uwagi są obliczane na podstawie podobieństwa między wektorem zapytania a wektorem klucza zgodnie z Równaniem 1117:

Tutaj dk oznacza wymiar wektora klucza. Ulepszona reprezentacja cech jest obliczana poprzez pomnożenie wag uwagi przez macierz wartości zgodnie z Równaniem 12:

Tutaj Fa stanowi ulepszoną reprezentację cech mapy cech. Ulepszona reprezentacja cech jest obliczana poprzez połączenie oryginalnych cech segmentacji z cechami ulepszonymi, uzyskanymi za pomocą mechanizmu uwagi zgodnie z Równaniem 13:
Tutaj Fe to ulepszona mapa cech wykorzystywana do rekonstrukcji wzorca. Moduł CAFFM został rozszerzony o wieloskalowy mechanizm wzajemnej uwagi (cross-attention), aby wyodrębnić cechy motywów kulturowych w różnych skalach. Niech Fsi oznacza cechy segmentacji w skali i, natomiast Faj oznacza cechy stylu artystycznego w tej samej skali. Końcowa reprezentacja cech jest zdefiniowana zgodnie z Równaniem 14:

Tutaj Qi, Ki i Vi reprezentują odpowiednio macierze zapytań (query), kluczy (key) i wartości (value) w skali i, a N oznacza liczbę skal cech. W niniejszym badaniu N = 4, co odpowiada mapom cech wyekstrahowanym w rozdzielczościach przestrzennych 1/4, 1/8, 1/16 i 1/32 rozmiaru obrazu wejściowego. Te wieloskalowe reprezentacje cech zostały połączone przy użyciu uczących się wag uwagi przed rekonstrukcją i syntezą artystyczną. Powyższe rozszerzenie umożliwia metodzie ekstrakcję globalnych motywów kulturowych i tekstur w celu rekonstrukcji wzorów kulturowych. Moduł CAFFM znajduje się pomiędzy etapem segmentacji opartym na SegFormer a etapem syntezy kreatywnej opartym na SPADE w proponowanym systemie SegCycle-SPADE. Najpierw, podczas gdy CycleGAN jednocześnie tworzy cechy rekonstrukcyjne z informacjami stylistycznymi i dotyczącymi wzorów, SegFormer-B2 odzyskuje semantyczne mapy cech, które opisują właściwości strukturalne motywów kulturowych. Moduł CAFFM przyjmuje te dwa strumienie cech i wykorzystuje fuzję opartą na mechanizmie cross-attention, aby zidentyfikować zależności między reprezentacjami strukturalnymi a artystycznymi. W celu stworzenia kulturowo autentycznych wzorów przy zachowaniu spójności semantycznej i cech strukturalnych, wynikowe połączone mapy cech są następnie przesyłane do modułu SPADE w celu syntezy kreatywnej sterowanej segmentacją.
Rekonstrukcja wzorów z wykorzystaniem CycleGAN
Po zakończeniu etapu wzmacniania cech opartego na mechanizmie uwagi, mapy cech będą zawierać wzmocnione struktury motywów kulturowych. Niemniej jednak mapy cech mogą nadal zawierać niekompletne lub uszkodzone obszary wzorów. W związku z tym, aby rozwiązać problem rekonstrukcji wzorów, w proponowanym schemacie wykorzystany zostanie model CycleGAN. W proponowanym schemacie dwiema domenami będą oryginalne wzory motywów kulturowych oraz zrekonstruowane wzory motywów kulturowych. Wykorzystując wzmocnione cechy z poprzednich etapów, model CycleGAN zrekonstruuje wzory kulturowe przy jednoczesnym zachowaniu spójności strukturalnej. Zapewni to, że wzory kulturowe, takie jak wzory geometryczne, roślinne i symboliczne, zachowają swój układ przestrzenny. Oryginalne obrazy batiku ludu Miao zostały poddane operacjom losowego maskowania i częściowego przesłaniania w celu wygenerowania niekompletnych lub uszkodzonych motywów kulturowych. Procedury degradacji te symulowały brakujące obszary wzorów i uszkodzenia strukturalne powszechnie obserwowane w zniszczonych artefaktach dziedzictwa kulturowego. Obrazy zdegradowane posłużyły jako dane wejściowe dla modułu rekonstrukcji, natomiast odpowiadające im obrazy oryginalne służyły jako obrazy referencyjne do oceny wydajności i jakości rekonstrukcji. Strategia ta umożliwiła modelowi naukę przywracania brakujących struktur motywów przy jednoczesnym zachowaniu spójności semantycznej i charakterystyki kulturowej.
Przyjmijmy, że X jest domeną niepełnych wzorców kulturowych, a Y domeną zrekonstruowanych wzorców kulturowych. Dwa generatory uczą się wykonywać mapowanie dwukierunkowe zgodnie z Równaniem 15:

Tutaj GE służy do rekonstrukcji struktur motywów, a FM do mapowania wzorców z powrotem na oryginalną domenę. Stratą przeciwniczą (adversarial loss) zapewnia się, że zrekonstruowane wzorce są realistyczne (Równanie 16)30

Tutaj DY jest dyskryminatorem używanym do rozróżnienia wzorców rzeczywistych i zrekonstruowanych, a GE(x) oznacza wygenerowany wzorzec zrekonstruowany. CycleGAN wymusza również spójność cyklu w celu zachowania układu strukturalnego motywów kulturowych (Równanie 17)30.

Końcowa funkcja straty została zdefiniowana przy użyciu Równania 1830:

Tutaj λi oznacza współczynnik wagowy dla straty spójności cyklu i został ustawiony na 10 podczas treningu, zgodnie z oryginalną sformułowaniem CycleGAN. Wartość ta została wybrana w celu zrównoważenia uczenia przeciwstawnego i spójności rekonstrukcji między domeną źródłową a docelową.
Moduł rekonstrukcji CycleGAN składa się z dwóch generatorów i dwóch dyskryminatorów służących do dwukierunkowej translacji obrazów. Każdy generator oparty jest na architekturze sieci rezdualnej, obejmującej początkową warstwę splotu 7 × 7, po której następują dwie warstwy splotu z próbkowaniem w dół, dziewięć bloków rezdualnych oraz dwie warstwy próbkowania w górę. Wszystkie operacje splotu wykorzystują jądro o rozmiarze 3 × 3, z wyjątkiem warstwy wejściowej, w której zastosowano jądro 7 × 7 w celu usprawnienia ekstrakcji cech. Po każdej warstwie splotu zastosowano normalizację instancji (Instance Normalization) w celu poprawy stabilności treningu, natomiast w całej sieci generatora wykorzystano funkcję aktywacji ReLU. Warstwa wyjściowa wykorzystuje funkcję aktywacji Tanh do generowania znormalizowanych obrazów wyjściowych. Dyskryminator opiera się na architekturze PatchGAN 70 × 70, składającej się z serii warstw splotowych o rozmiarach jądra 4 × 4 i sukcesywnie zwiększającej liczbie kanałów cech. W dyskryminatorze zastosowano normalizację instancji oraz aktywację LeakyReLU (ujemny nachylenie = 0.2), aby poprawić dyskryminację cech i uczenie przeciwstawne. Moduł CycleGAN przyjmuje jako wejście przetworzone obrazy motywów kulturowych i generuje zrekonstruowane reprezentacje wzorów, które są następnie przekazywane do CAFFM w celu integracji z cechami segmentacji. Trening CycleGAN przeprowadzono przy użyciu optymalizatora Adam (β₁ = 0.5, β₂ = 0.999) z początkową prędkością uczenia 0.0002. Prędkość uczenia utrzymywano przez pierwsze 100 epok, a następnie liniowo zmniejszano do zera w pozostałym okresie treningu. W celu stabilizacji treningu dyskryminatora zastosowano bufor powtórek (replay buffer) zawierający 50 wcześniej wygenerowanych obrazów. Aktualizacja generatorów i dyskryminatorów odbywała się w stosunku 1:1, przy czym jednej aktualizacji generatora towarzyszyła jedna aktualizacja dyskryminatora podczas każdej iteracji treningowej.
Proces rekonstrukcji w sieci CycleGAN opiera się na ulepszonych mapach cech uzyskanych za pomocą mechanizmu CAFFM przedstawionego na Rysunku 5. Mapy cechy zawierają wzmocnione motywy kulturowe otrzymane w poprzednich etapach segmentacji oraz CAFFM. Mapy cech służą jako dane wejściowe dla pierwszego generatora GE. Pierwszy generator GE uczy się odwzorowania niezbędnego do rekonstrukcji wzorców kulturowych. Wyniki są następnie przekazywane do dyskryminatora DY, aby odróżnić rzeczywiste wzorce kulturowe od wzorców zrekonstruowanych. Dyskryminator DY pomaga generatorowi GE generować realistyczne wyniki. Aby zapewnić, że wzorce kulturowe nie zostaną zniekształcone podczas rekonstrukcji, drugi generator FM wykonuje odwzorowanie spójności cyklu (cycle-consistency mapping). Drugi generator FM mapuje wyniki z powrotem do domeny oryginalnej. Wyniki te są następnie oceniane przez dyskryminator w celu zapewnienia realizmu. Końcowe wyniki są następnie przekazywane do modułu SPADE w celu generowania stylu artystycznego w kolejnym etapie proponowanej architektury SegCycle-SPADE.

Rycina 5. Schemat rekonstrukcji wzorów oparty na CycleGAN. Przebieg pracy modułu rekonstrukcji CycleGAN. Reprezentacje cech wzmocnione mechanizmem uwagi są dostarczane jako dane wejściowe do sieci generatora w celu rekonstrukcji niepełnych motywów kulturowych. Dyskryminator ocenia zrekonstruowane wyniki w odniesieniu do wzorców referencyjnych, podczas gdy mapowanie spójności cyklu zachowuje integralność strukturalną podczas dwukierunkowej translacji obrazów. Zrekonstruowane motywy są następnie przekazywane do modułu SPADE w celu syntezy stylu artystycznego. Kliknij tutaj, aby wyświetlić większą wersję tej ryciny.
Generowanie stylu artystycznego za pomocą SPADE
Następnie zrekonstruowane motywy kulturowe są poddawane działaniu modułu SPADE w celu generowania stylu artystycznego. Głównym celem modułu SPADE jest dodanie do zrekonstruowanych motywów kulturowych bogatszych wizualnie tekstur stylu artystycznego, bez naruszania układu strukturalnego motywów. Moduł SPADE jest techniką warunkowego generowania obrazów, która wykorzystuje koncepcję segmentacji obrazu do tworzenia grafik. Mapy semantyczne wielokanałowe, odpowiadające określonym klasom motywów, zostały zakodowane na podstawie masek segmentacji semantycznej wygenerowanych przez SegFormer-B2. Generator SPADE otrzymał te zakodowane mapy jako wejścia warunkujące. Przestrennie adaptacyjne parametry skalowania (γ) i przesunięcia (β) były wytwarzane poprzez przetwarzanie map semantycznych w warstwach splotowych w obrębie każdej warstwy SPADE. Dzięki zastosowaniu tych parametrów do znormalizowanych aktywacji cech, generator był w stanie zachować granice motywów, układy strukturalne oraz informacje semantyczne podczas syntezy artystycznej. Naprowadzanie semantyczne mogło wpływać zarówno na wysokopoziomową rekonstrukcję strukturalną, jak i niskopoziomową syntezę tekstury, ponieważ proces warunkowania odbywał się na kilku warstwach generatora SPADE. W rezultacie stworzone wzory artystyczne zawierały cechy stylistyczne pozyskane z zestawu danych WikiArt, przy jednoczesnym zachowaniu struktur motywów kulturowych zidentyfikowanych na etapie segmentacji.
Zbiór danych WikiArt, obejmujący prace z 27 różnych kategorii artystycznych — takich jak renesans, impresjonizm, kubizm, ekspresjonizm, surrealizm, realizm i sztuka abstrakcyjna — posłużył jako główny zasób do zrozumienia stylów artystycznych. Aby zapoznać model z różnorodnymi cechami twórczymi i zwiększyć generalizację stylu, podczas trenowania losowo wybierano obrazy stylistyczne z poszczególnych kategorii. Zbiór danych podzielono na podzbiory treningowy, walidacyjny i testowy z zrównoważoną reprezentacją kategorii artystycznych, aby zminimalizować błąd systematyczny związany ze stylem. W celu zapewnienia reprezentatywności wszystkich 27 kategorii artystycznych zastosowano próbkowanie warstwowe. Próbki z każdej kategorii zostały proporcjonalnie przydzielone do podzbiorów treningowego, walidacyjnego i testowego przy zachowaniu pierwotnego rozkładu klas. Moduł CAFFM wykorzystano do połączenia aspektów stylu wywodzących się z obrazów WikiArt z cechami rekonstrukcyjnymi wygenerowanymi przez CycleGAN. Aby umożliwić sieci syntezy przeniesienie cech artystycznych przy jednoczesnym zachowaniu struktury semantycznej i granic motywów kulturowych wynikających z map segmentacji, otrzymane reprezentacje połączone przekazano jako informacje warunkujące do generatora SPADE. Dzięki tej technice warunkowania stylu zaproponowana struktura była w stanie generować kulturowo autentyczne wzory artystyczne o spójnych reprezentacjach strukturalnych i stylistycznych.
SPADE wprowadza również parametry adaptacyjne przestrzennie, które zależą od mapy segmentacji. Parametry te można zdefiniować zgodnie z Równaniem 191:

Tutaj γ(S) jest przestrzennie zmiennym parametrem skali, a β(S) jest przestrzennie zmiennym parametrem przesunięcia. Parametry te mogą pomóc generatorowi w tworzeniu różnych tekstur i stylów w zależności od regionu semantycznego na obrazach. Końcowe dzieło sztuki kulturowej można zdefiniować zgodnie z Równaniem 20:

W tym przypadku GE jest generatorem SPADE, XrP jest zrekonstruowanym wzorem, a SM to mapa segmentacji. Końcowe dzieło zachowuje integralność strukturalną motywów kulturowych, jednocześnie poprawiając walory artystyczne. W celu optymalizacji zaproponowanej architektury SegCycle-SPADE zastosowano złożoną funkcję straty, która równoważy dokładność segmentacji semantycznej, zachowanie motywów kulturowych, jakość rekonstrukcji wzoru oraz spójność stylu artystycznego. Całkowity cel treningowy ustalono jako ważoną mieszaninę straty segmentacji (Lseg), straty adwersarialnej (Ladv), straty spójności cyklu (Lcycle), straty rekonstrukcji (Lrecon), straty zachowania motywu (Lmotif) oraz straty spójności stylu (Lstyle). Całkowita funkcja straty została zdefiniowana w Równaniu 21:
(21)
Aby zapewnić spójność strukturalną między motywami kulturowymi wejściowymi a zrekonstruowanymi, współczynnik straty spójności cyklicznej ustawiono na 10. W celu zachowania szczegółowych cech motywów i zwiększenia dokładności wizualnej, współczynnik straty rekonstrukcji ustawiono na 5. Aby podkreślić zachowanie istotnych kulturowo wzorców strukturalnych podczas syntezy artystycznej, dla straty zachowania motywu zastosowano współczynnik 2. W celu promowania transferu stylu artystycznego bez nadmiernego zniekształcania semantycznych struktur motywów, współczynnik straty spójności stylu dostosowano do 1. Aby utrzymać zrównoważony wkład pomiędzy generowaniem realistycznych obrazów a precyzyjną segmentacją motywów, przypisano równe wagi stracie segmentacji oraz stracie adwersarialnej. Do obliczenia straty spójności stylu wykorzystano reprezentacje stylu oparte na cechach z zestawu danych WikiArt. W szczególności cechy stylu artystycznego uchwycono za pomocą korelacji macierzy Grama wyliczonych z głębokich map cech. Do obliczenia straty stylu wykorzystano różnicę normy Frobeniusa między macierzami Grama obrazu stylu docelowego a obrazu wygenerowanego, co przedstawiono w Równaniu 22:

Tutaj, Gl czy macierz Grama jest obliczana na podstawie lth warstwa cech, Jagen oznacza wygenerowany obraz artystyczny, Jastyl oznacza obraz stylu docelowego z zestawu danych WikiArt, a F oznacza normę Frobeniusa. Takie sformułowanie pozwala proponowanemu frameworkowi zachować cechy artystyczne przy jednoczesnym utrzymaniu integralności strukturalnej i semantycznej motywów kulturowych.
Połączone reprezentacje cech wytworzone przez moduł CAFFM są wykorzystywane jako wejścia warunkujące dla modułu SPADE, który pełni funkcję komponentu syntezy artystycznej w proponowanym schemacie. Generator SPADE składa się z siedmiu rezidualnych bloków SPADE o latentnym wymiarze cech wynoszącym 256 kanałów i generuje obrazy wyjściowe o rozdzielczości 256 × 256 pikseli. Mapy segmentacji semantycznej uzyskane z modułu SegFormer–CAFFM są wykorzystywane jako wejścia warunkujące w całej strukturze rezidualnych warstw SPADE. Warstwy SPADE są stosowane przed funkcjami aktywacji w każdym bloku rezidualnym, co umożliwia semantyczne warunkowanie sterowane segmentacją. Poprzez następujące operacje nadpróbkowania i splotu, latentna reprezentacja cech jest sukcesywnie dopracowywana w celu wygenerowania artystycznych wzorów o wysokiej rozdzielczości przy jednoczesnym zachowaniu spójności semantycznej i struktury motywu. W całym generatorze stosowane są funkcje aktywacji LeakyReLU, a w warstwie wyjściowej zastosowano funkcję aktywacji Tanh w celu uzyskania znormalizowanych obrazów.
Algorytm i schemat pracy
Struktura SegCycle-SPADE integruje segmentację semantyczną, fuzję cech, rekonstrukcję wzorów oraz syntezę stylu artystycznego w ramach zunifikowanego potoku treningowego. Schemat pracy rozpoczyna się od pozyskania i wstępnego przetwarzania zbioru danych, obejmującego zmianę rozmiaru obrazów, normalizację, odszumianie oraz przygotowanie masek segmentacyjnych. Przetworzone wstępnie obrazy są następnie analizowane przez sieć segmentacyjną SegFormer-B2 w celu wyodrębnienia semantycznych reprezentacji motywów. Uzyskane cechy segmentacji są łączone z cechami rekonstrukcji poprzez moduł CAFFM, co umożliwia interakcję między informacjami o motywach strukturalnych a reprezentacjami cech artystycznych. Zfuzowane mapy cech są następnie dostarczane do modułu rekonstrukcji CycleGAN, który odtwarza niekompletne struktury motywów kulturowych przy zachowaniu spójności semantycznej. Zrekonstruowane wzory są następnie przekazywane do generatora SPADE w celu syntezy artystycznej sterowanej segmentacją, co pozwala na ulepszenie stylistyczne przy jednoczesnym zachowaniu granic motywów i autentyczności strukturalnej. Podczas treningu parametry modelu są optymalizowane za pomocą złożonej funkcji straty opisanej w Równaniach 21 i 22, która balansuje dokładność segmentacji, zachowanie motywów, jakość rekonstrukcji oraz spójność stylu artystycznego. Szczegółowy, krok po kroku opisany schemat implementacji, obejmujący ładowanie zbioru danych, wstępne przetwarzanie, inicjalizację modelu, iteracje treningowe, procedury walidacji, wybór punktów kontrolnych oraz końcową ewaluację, znajduje się w Pliku uzupełniającym 1 (Algorytm 1). Kompletny schemat algorytmiczny został zaimplementowany przy użyciu rozmiaru batcha wynoszącego 16, współczynnika uczenia 0.0002 oraz 100 epok treningowych. Stałe ziarno losowe 42 zostało zastosowane do partycjonowania zbioru danych, inicjalizacji modelu oraz wszystkich eksperymentów treningowych. Ziarno to było konsekwentnie stosowane w generatorach liczb losowych Python, NumPy i PyTorch w celu zapewnienia powtarzalności. Optymalizator Adam skonfigurowano z parametrami β₁ = 0.5, β₂ = 0.999 oraz bez rozkładu wag (weight decay = 0). Punkty kontrolne modelu wybrano na podstawie najwyższej uzyskanej wartości IoU na zbiorze walidacyjnym.
Optymalizacja funkcji straty
Aby zachować struktury motywów kulturowych podczas rekonstrukcji i syntezy artystycznej, proponowana struktura wykorzystuje złożony cel optymalizacji, który łączy straty segmentacji, adwersarialne, spójności cyklicznej, rekonstrukcji, zachowania motywów oraz spójności stylu. Pełna formuła matematyczna, współczynniki wagowe oraz definicja straty stylu zostały przedstawione w Równaniach 21 i 22 w podrozdziale Artistic Style Generation using SPADE. Komponent zachowania motywów penalizuje odchylenia strukturalne między przewidywanymi obszarami motywów a odpowiadającymi im maskami segmentacji prawdy obiektywnej (ground-truth), co sprzyja zachowaniu istotnych kulturowo struktur wzorów w całym procesie rekonstrukcji.