Improving AI-based Amplified P-wave Morphology Detection