Restoring Without Forgetting: Continual Learning Across Image Degradations
Dit artikel introduceert Restoring without Forgetting (RwF), een framework voor continual learning dat lichtgewicht adapters en unsupervised routing gebruikt om beeldherstelmodellen in staat te stellen sequentieel nieuwe degradatietypes te leren zonder catastrofale vergetelheid of toegang tot historische data, waarbij het standaard fine-tuning benaderingen aanzienlijk overtreft.
Oorspronkelijk artikel gelicentieerd onder CC BY 4.0 (http://creativecommons.org/licenses/by/4.0/). Dit is een AI-gegenereerde uitleg van het onderstaande artikel. Het is niet geschreven of goedgekeurd door de auteurs. Raadpleeg het oorspronkelijke artikel voor technische nauwkeurigheid. Lees de volledige disclaimer
Stel je een camerasensor voor die haar hele leven in een zonnige, heldere woestijn heeft doorgebracht. Ze heeft perfect geleerd hoe ze de wereld onder dat specifieke licht moet zien. Stel je nu voor dat diezelfde camera wordt verplaatst naar een regenachtige stad, en daarna naar een mistige berg, en tot slot naar een donkere grot. In de echte wereld worden sensoren geconfronteerd met deze veranderende omstandigheden, één na het andere, vaak zonder toegang tot de oude gegevens van vorige locaties vanwege privacyregels of opslaglimieten. Om een computer te leren hoe hij afbeeldingen uit een nieuwe omgeving moet herstellen, moet hij meestal vanaf nul opnieuw worden getraind. Als je probeert het systeem de nieuwe omstandigheden te leren zonder het geheugen van de oude omgeving te wissen, vergeet het vaak volledig hoe het de eerste omgeving moet verwerken. Dit is een grote hindernis voor het bouwen van intelligente systemen die in de loop van de tijd kunnen adapteren. Het vakgebied van beeldrestauratie, dat erop gericht is wazige, ruisgevoelige of donkere foto's op te schonen, heeft grotendeels vertrouwd op gespecialiseerde hulpmiddelen voor elk specifiek probleem, of op enorme "all-in-one"-systemen die toegang vereisen tot elk type probleem tegelijkertijd om te kunnen leren. Geen van beide benaderingen werkt goed wanneer een systeem sequentieel moet leren, waarbij het nieuwe uitdagingen één voor één tegenkomt terwijl het zijn oude vaardigheden intact houdt.
Onderzoekers aan het Rochester Institute of Technology hebben een nieuwe methode ontwikkeld genaamd "Restoring Without Forgetting" om exact dit probleem op te lossen. In plaats van voor elk nieuw type beeldbeschadiging een enorm nieuw netwerk te trainen, of te proberen al het leren in één gigantisch model te proppen dat uiteindelijk in de war raakt, hebben zij een systeem gebouwd dat leert adapteren zonder het verleden te verliezen. De kern van hun aanpak is een vooraf getraind "backbone"-netwerk dat al erg goed is in het zien van schone afbeeldingen. Dit backbone-netwerk is bevroren, wat betekent dat de interne instellingen vergrendeld zijn en nooit worden gewijzigd. Wanneer het systeem een nieuw type degradatie tegenkomt, zoals regen of mist, traint het niet het hele brein opnieuw. In plaats daarvan koppelt het een kleine, lichtgewicht add-on module aan dat specifiek is ontworpen voor die nieuwe conditie. Deze add-on leert alleen de specifieke patronen van het nieuwe probleem, waardoor de oorspronkelijke kennis van het backbone-netwerk en de vorige add-ons volledig onaangetast blijven. Omdat elke nieuwe vaardigheid zijn eigen, geïsoleerde ruimte krijgt, overschrijft het systeem de eerder geleerde kennis nooit.
Om dit in de echte wereld werkend te maken, waar de computer niet weet met welk type schade hij te maken heeft, hebben de onderzoekers een slim routingmechanisme toegevoegd. Wanneer een nieuwe, beschadigde afbeelding binnenkomt, scant het systeem deze kort om een eenvoudige handtekening te creëren op basis van visuele aanwijzingen. Vervolgens vergelijkt het deze handtekening met een kleine bibliotheek van opgeslagen voorbeelden voor elk bekend type degradatie. Op basis van welk voorbeeld het het dichtstst overeenkomt, selecteert het systeem automatisch de juiste add-on om de afbeelding te herstellen. Dit gebeurt zonder dat een mens het systeem vertelt wat het probleem is. De onderzoekers testten dit op een opeenvolging van vijf verschillende beeldproblemen: ruis, bewegingsonscherpte, regen, nevel en lichtomstandigheden met weinig licht. Ze ontdekten dat een standaardmethode van het simpelweg opnieuw trainen van het netwerk op nieuwe gegevens ervoor zorgde dat het systeem de vorige vaardigheden zo slecht vergat dat de prestaties op eerdere taken met meer dan 15 punten daalden op een standaard kwaliteitschaal. In contrast hiermee behield hun nieuwe methode een hoge prestatie over alle vijf de taken, waardoor het vergeetprobleem effectief volledig werd geëlimineerd.
De studie toonde ook aan dat deze aanpak werkt op echte afbeeldingen, en niet alleen op synthetische beelden die in een laboratorium zijn gemaakt. Het team heeft hun systeem getest op elf verschillende real-world datasets die duizenden afbeeldingen bevatten met natuurlijke onscherpte, regen en andere imperfecties. Het systeem identificeerde bijna 90 procent van de afbeeldingen correct als het juiste reparatietool, en de uiteindelijke beeldkwaliteit was bijna net zo goed als wanneer een mens het systeem precies had verteld waarmee het te maken had. Dit suggereert dat het systeem kan generaliseren naar nieuwe, ongeziene inhoud, zolang het type schade maar vergelijkbaar is met wat het eerder heeft gezien. De onderzoekers stellen dat dit een belangrijke stap voorwaarts is, omdat het een praktische manier vestigt voor beeldrestauratiesystemen om in de loop van de tijd te groeien en te adapteren zonder de noodzaak om enorme hoeveelheden oude gegevens op te slaan of vanaf nul te beginnen met trainen. Door het leren van nieuwe vaardigheden te isoleren van de kernkennis van het systeem, hebben zij een raamwerk gecreëerd dat kunstmatige intelligentie in staat stelt om capaciteiten sequentieel te accumuleren, net zoals een mens nieuwe vaardigheden leert zonder de oude te vergeten.
Verdrinkt u in papers in uw vakgebied?
Ontvang dagelijkse digests van de nieuwste papers die bij uw onderzoekswoorden passen — met technische samenvattingen, in uw taal.