A tri-partite modular autoencoder for addressing imbalanced cell type annotation and batch effect reduction