Tag neural network architecture