Aaron Bateni

I am an undergraduate researcher at University of Tehran, part of the Dr. Bagher Babaali's lab, where we're working on tweaking transformer models to classify emotions effectively.

Previously, I had the privilege of working under the supervision of Dr. Mohammad Ganjtabesh on designing a first-of-its-kind neural architecture for time-series data classification (mainly video) as my BSc thesis (See Below). Before that, I completed a research internship at Institute for Research in Fundamental Sciences (IPM) one of Iran's leading research institutes, where I worked on utilizing transformer models to segment smoke as a fire detection system.

I hold a BSc in Computer Science from University of Tehran, graduating Summa Cum Laude and ranking 2nd in my cohort.

CV CV   |   Email Email   |   GitHub GitHub   |   Google Scholar Google Scholar   |  
LinkedIn LinkedIn   |   YouTube YouTube Talks

profile photo

Research

My research interest is in developing novel methods and optimization techniques that enable and improve intelligent reasoning within different problem domains. Aiming one day, to make people utilize models like LLMs, ViTs, or even diffusion models like DiTs to solve current tasks, or innovative new tasks efficiently on day-to-day edge devices (e.g. phones and laptops).

project image

Multi-Class Emotion Classification Using Transformers (In progress)


Aaron Bateni, Ali Abdollahi Asl, Bagher Babaali
Coming Soon, 2025

WESAD dataset is a mix of time-series biological signals (such as ECG, EDA, EMG, respiration, etc.) used for emotion classification. While majority of previous works focused on combining all classes to binary classification, we introduce a new method to classify multiple classes. The model Achieves an F1-Score of 0.93, which to the best of our knowledge is the State-Of-The-Art.

project image

Hybrid Spiking Neural Network -- Transformer Video Classification Model


Aaron Bateni, Mohammad Ganjtabesh
arXiv, 2024
PDF | arXiv | Code |

Inspired by hybrid Transformer-CNN models, we design a first-of-its-kind hybrid Transformer-SNN model that is inspired by the Cortical Column structure of the brain. The model is then implemented and tested in time-series classification. This work was a result of my original BSc thesis.




Projects

I have also worked on a variety of personal projects. some of which are featured below. for a full list, I recommend checking out my GitHub page.

project image

Study on Effects of UNet's Variations in Polyp Segmentation


University of Tehran
2024
Page |

I conducted a study on effects of architectural tweaks in the UNets model architecture. I was particularly interested to see how the height of the UNet affects its performance.

project image

Loss Function Optimization with Particle Swarm Optimization


University of Tehran
2024
Page |

I designed a PSO method that minimizes a given loss function by deploying searching particles with momentum and density.

project image

Pacman Playing Agent


University of Tehran
2024
Page |

I developed a Q-Learning agent that learns and plays the Pacman game.

project image

Implementation of Spiking Neural Architectures


University of Tehran
2024
Page |

Implemented several Neural Architectures as prerequisites to my BSc Project. The list is as follows: (Image Credit: Wikimedia)

  1. Neural Models: LIF, ELIF, AELIF
  2. Encoding Methods: Poisson, TTFS, Positional
  3. Currents: Sinusoidal, Constant, Step Current, Multi Step Current

Extensive hands-on experience with Pytorch.

project image

Comprehensive Analysis of Information Retrieval Algorithms


University of Tehran
2023
Page #1 | Page #2 | Page #3 | Page #4 | Page #5 |

Comprehensive implementations of several Information Retrieval Systems. The list is as follows:

  1. WebIR with boolean queries
  2. WebIR with wildcard query (Trie, and Permuterm), and spellchecking (Soundex, and Levenshtein Distance)
  3. WebIR with Block-Sorted Based Indexing
  4. Ranked WebIR with sorting queries based on relevance (Okapi BM25, Language Model, and TF-IDF)
  5. Comprehensive Performance Comparison of IR methods
project image

Review of Ant Colony Methods in SCP


University of Tehran
2022
Page |

Implementations and comparison of several Ant Colony System methods to solve the Set Covering Problem. (Image credit: Wikimedia)

project image

Simulated Annealing for CSP


University of Tehran
2022
Page |

I designed a simulated annealing method to solve the classic Cutting Stock Problem.

project image

Genetic Method for N-Queen Problem


University of Tehran
2022
Page |

I developed an evolutionary genetic algorithm to solve the classic N-Queens problem.

project image

ClassBoom


University of Tehran
2021
Page |

Effected by the boom of E-Learning during the Covid-19 pandemic, I set out to create an E-Learning platform from scratch. Some features include the ability to upload class recordings, homework, grade students individually and more. I mainly utilized Python and Django.

project image

Corridor Game


University of Tehran
2021
Page |

Using C++, I developed a client-server program to play the world famous Quoridor Game on lan. (Image credit: Wikimedia)




Teaching & Academic Service

I have also been an active member of our academic community. Some of my main contributions are as follows.

project image

Teaching Assistantship


University of Tehran
2021 - 2024

  • TAed for: Design and Analysis of Algorithms, Graph Theory, Logical Circuits & Architecture, Combinatorics, Calculus I, and Basic Programming.
  • Held classes for 250+ students and prepared 25+ assignment series.
  • Designed real-world problems and aggregated widely used problems from academic references.
  • Graded 1000+ student submissions, quizzes and exams.
project image

Elected Member of the Computer Science Students’ Scientific Chapter


University of Tehran
2021 - 2023

  • Organized workshops on career development, resume writing, and programming with distinguished speakers.
  • Organized orientation for the new students on various topics such as programming languages/disciplines.

Design and source code partially borrowed from Jon Barron's website