SHAPE AWARENESS AND INTERPRETABILITY IN DEEP NETWORKS USING GEOMETRIC MOMENTS

A Deep Geometric Moment framework (DGM framework) may be trained to perform an image classification task using a training dataset having a plurality of images. Training the DGM framework may include generating 2D coordinate grids for the plurality of images and computing coordinate bases. The training may further include extracting image features for the plurality of images and computing geometric moments for the image features. Training the DGM framework may include generating predicted affine transformation parameters to transform the 2D coordinate bases into a transformed coordinate grid and generating new coordinate bases and new geometric moments learned from the image features to reconstruct the plurality of images using the transformed coordinate grid. The DGM framework may output an Artificial Intelligence model (AI model) trained to perform the image classification task for an input image.