---
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