CBGT-Net: A Neuromimetic Architecture for Robust Classification of Streaming Data

June 2024

CBGT-Net: A Neuromimetic Architecture for Robust Classification of Streaming Data

Authors:

Shreya Sharma

Abstract:

This research introduces CBGT-Net, a neural network model inspired by the cortico-basal ganglia-thalamic (CBGT) circuits in mammalian brains, which are crucial for critical thinking and decision-making. Unlike traditional neural network models that generate an output for each input or after a fixed sequence of inputs, CBGT-Net learns to produce an output once sufficient evidence for action is accumulated from a stream of observed data. For each observation, CBGT-Net generates a vector representing the amount of evidence for each potential decision, accumulates this evidence over time, and makes a decision when the accumulated evidence surpasses a predefined or dynamically learned threshold.

We evaluate the proposed model on various image classification tasks, where models must predict image categories based on a stream of partially informative visual inputs. Our results demonstrate that CBGT-Net offers improved accuracy and robustness compared to models trained to classify from a single image, as well as models utilizing an LSTM layer or a ViT-style transformer to classify from a fixed sequence of image inputs. Additionally, we introduce a novel dataset for classification based on sequential image data of urban city buildings. This dataset provides multi-view images of 3D building assets on fire, categorized into five stages of fire severity.

Notes:

@mastersthesis{Sharma-2024-141266,
author = {Shreya Sharma},
title = {CBGT-Net: A Neuromimetic Architecture for Robust Classification of Streaming Data},
year = {2024},
month = {June},
school = {Carnegie Mellon University},
address = {Pittsburgh, PA},
number = {CMU-RI-TR-24-29},
keywords = {Neuro-inspired, Machine Learning, CBGT, Decision-making, Image classification, Evidence accumulation, Cortico-basal ganglia-thalamic circuits, Sequential image data},
}
Copyright notice: This material is presented to ensure timely dissemination of scholarly and technical work. Copyright and all rights therein are retained by authors or by other copyright holders. All persons copying this information are expected to adhere to the terms and constraints invoked by each author's copyright. These works may not be reposted without the explicit permission of the copyright holder.