AI in Service Assurance for Critical Communications