| --- |
| tags: |
| - custom |
| - cifar-10 |
| - image-classification |
| - block-architecture |
| language: en |
| framework: pytorch |
| metrics: |
| - accuracy: 75.43 |
| license_name: mit |
| datasets: |
| - CIFAR-10 |
| --- |
| |
| # BlockNet10 - CNN for CIFAR-10 dataset |
|
|
| ## Overview |
|
|
| BlockNet10 is a neural network architecture designed for image classification tasks using the CIFAR-10 dataset. This model implements a sequence of intermediate blocks (B1, B2, ..., BK) followed by an output block (O). |
|
|
| ## Architecture Details |
|
|
| ### Intermediate Block (Bi) |
|
|
| Each intermediate block receives an input image x and outputs an image x'. The block comprises L independent convolutional layers, denoted as C1, C2, ..., CL. |
|
|
| Each convolutional layer Cl in a block operates on the input image x and outputs an image Cl(x). |
|
|
| <div style="display: flex; justify-content: center;"> |
| <img src="figures/eq1.png" alt="Equation 1" /> |
| </div> |
|
|
| The output image x' is computed as x' = a1C1(x) + a2C2(x) + ... + aLCL(x), where a = [a1, a2, ..., aL]T is a vector computed by the block. |
|
|
| The vector a is obtained by computing the average value of each channel of x and passing it through a fully connected layer with the same number of units as convolutional layers in the block. |
|
|
| <div style="display: flex; justify-content: center;"> |
| <img src="figures/fig1.png" alt="Figure 1" /> |
| </div> |
|
|
| ### Output Block (O) |
|
|
| The output block processes the final output image from the intermediate blocks for classification. |
|
|
| ## Analytics |
|
|
| <div style="display: flex; justify-content: center; align-items: center;"> |
| <table> |
| <tr> |
| <th>Epoch Number</th> |
| <th>Train Accuracy</th> |
| <th>Test Accuracy</th> |
| <th>Average Loss</th> |
| </tr> |
| <tr> |
| <td>50</td> |
| <td>75.43</td> |
| <td>80.56</td> |
| <td>0.685</td> |
| </tr> |
| </table> |
| </div> |
| |
| ## Clone on GitHub |
|
|
| You can contribute to the advancement of this architecture, changes in hyperparameter, or solve issues <a href="https://github.com/siddheshtv/cifar10" target="_blank">here</a>. |
|
|
| ## Citation |
|
|
| If you use BlockNet10 in your research or work, please cite it as follows: |
|
|
| ```bibtex |
| @article{blocknet10, |
| title={BlockNet10: CIFAR-10 Image Classifier}, |
| author={Siddhesh Kulthe}, |
| year={2024}, |
| publisher={Hugging Face}, |
| url={https://huggingface.co/siddheshtv/BlockNet10} |
| } |
| ``` |
|
|
| --- |
|
|
| ## license: mit |
|
|