The original paper that Rognlie wrote:

http://www.mit.edu/~mrognlie/piketty_diminishing_returns.pdf