If you would like to return additional values in the forward method, but keep the model at it was otherwise, you could directly derive your class from the base model:
class FeatureExtractedModel(BaseModel)

This would make the implementation cleaner in my opinion.