That actually sounds like a really fun feature. However most air attacks are designed to be used during a full horizontal airdash. The closest that might work for you is the Quick Drop badge. It allows you to interrupt jumps by pressing down and effectively doing KOF shorthops.