The Gemara itself does not seem to be referring to "Yesurim Shel Ahavah" -- "suffering of love" that you are referring to, which is endured by people who Hash-m wants to give a better portion in the next world, as their Mitzvos alone will not get them that portion (according to Rashi in Berachos 5a). The Ramban in Shar Ha'Gemul argues that "Yesurim Shel Ahavah" takes away small sins such as those done accidentally.

However, this Gemara (as explained by the Maharsha) is referring to suffering which takes away one's sins in this world so that he will not have to be punished in the next world. "Better in this world" is probably the practical phrase which applies to almost all afflictions nowadays. When a person realizes that Hash-m is punishing him because He does not want him to suffer in the next world, and accepts this with love, he is solidifying his relationship with Hash-m. A story is told regarding R' Avigdor Miller zt"l that when the doctor apologized to him that a certain procedure was going to cause him pain, he responded, "The more the better!" This is the meaning of our Gemara.

All the best,

Yaakov Montrose

Rebbi Akiva's afflictions can definitely be said to fall in the category of "Yesurim Shel Ahavah."