Skip to content
Snippets Groups Projects
README.md 1.91 KiB
Newer Older
  • Learn to ignore specific revisions
  • vlevente24's avatar
    vlevente24 committed
    # Neurális Hálózatok Összehasonlítása CIFAR-10 Adathalmazon
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    **Szerző:** Vajna Levente  
    **Utolsó frissítés:** 2025. május 6.
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ## Bevezetés
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    Ez a projekt három különböző neurális hálózati architektúra teljesítményét hasonlítja össze a CIFAR-10 képosztályozási feladaton:
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    - **CeNN (Cellular Neural Network)**
    - **CNN (Convolutional Neural Network, ResNet18 alapú)**
    - **Hibrid modell (CeNN + CNN kombináció)**
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    A **CIFAR-10** adathalmaz 60 000 db 32x32 pixeles színes képet tartalmaz 10 kategóriában, melyből 10 000 kép tesztelésre van fenntartva.
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ---
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ## Modellek
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ### CeNN (Cellular Neural Network)
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    - Lokális kapcsolatokra épülő architektúra  
    - `A`, `B`, `Z` paraméterek tanulása  
    - Euler-módszer az állapotegyenlet megoldására  
    - Bináris klasszifikáció: **0. osztály vs többi**
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ### CNN (Convolutional Neural Network)
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    - **ResNet18** architektúra, előre tanított súlyokkal  
    - CIFAR-10 specifikus módosítások:
      - 3×3-as konvolúciós magok
      - Max pooling eltávolítása  
    - 10 osztályos klasszifikáció
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ### Hibrid Modell
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    - **CeNN** előfeldolgozó réteg  
    - **ResNet18** osztályozó réteg  
    - Cél: mindkét modell előnyeinek kombinálása
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ---
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ## Eredmények
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    | Modell   | Tesztpontosság | Tanítási idő (GPU) |
    |----------|----------------|--------------------|
    | CeNN     | 90.02%         | ~20 perc           |
    | CNN      | 92.40%         | ~31 perc           |
    | Hibrid   | 91.45%         | ~26 perc           |
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ---
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    ## Következtetések
    
    Vajna Levente's avatar
    Vajna Levente committed
    
    
    vlevente24's avatar
    vlevente24 committed
    - A **CNN modell** érte el a legjobb tesztpontosságot (**92.40%**) viszonylag rövid tanítási idővel (~31 perc).
    - A **CeNN modell** a leggyorsabban tanítható modell volt (~20 perc), és így is versenyképes pontosságot ért el. (**90.02%**)
    - A **hibrid modell** teljesítménye köztes eredményt hozott (**91.45%**), gyorsabb volt a CNN-nél, de csak azért, mert annál minden epoch után ki is értékeltük.