Nonlinear geometric optics method-based multi-scale numerical schemes for a class of highly oscillatory transport equations