É possível limitar o tamanho dos lotes retornados pelo consumidor Kafka para o Spark Streaming?

Estou perguntando, porque o primeiro lote que recebo tem centenas de milhões de registros e leva muito tempo para processá-los e conferi-los.