Yes, this would be a good optimization. I started this but then found myself extracting stuff from MabBasedInputRow (getDimension, getFloatMetric) and the dimension exclusion stuff from MapBasedInputRowParser that will be needed by the new ProtoBufBasedInputRow and its parser. I will continuer this as soon as I have more time and will open another pull request for that later on.