2060260 - [Models] ATG Improve Clustering Algorithm

Status: VERIFIED FIXED

(Core :: Machine Learning: Frontend, task)

Description

[Vasish Baungally]

Current algorithm uses Kmeans. We previously tested agglomerative clustering against our test set and found quite some improvements there. Do some more evaluations and add method if metrics improve.

Comment 1

[Vasish Baungally]

Attachment: Bug 2060260 - Add agglomerative clustering algorithm for Auto Tab Grouping. r?rrando,jlewis

This patch switches the KMeans clustering algorithm used for Auto Tab Grouping to
Agglomerative clustering. The latter is more performant and has better metrics on our
golden dataset. It is also deterministic.

Comment 2

[Pulsebot]

Pushed by [e-mail]:
https://github.com/mozilla-firefox/firefox/commit/7d249b9965ab
https://hg.mozilla.org/integration/autoland/rev/35ce7d711347
Add agglomerative clustering algorithm for Auto Tab Grouping. r=rrando,jlewis,tabbrowser-reviewers,ai-platform-reviewers,dao,valentinp

Comment 3

[Vasish Baungally]

Attachment: Bug 2060260 - Add agglomerative clustering algorithm for Auto Tab Grouping.

This patch switches the KMeans clustering algorithm used for Auto Tab Grouping to
Agglomerative clustering. The latter is more performant and has better metrics on our
golden dataset. It is also deterministic.

Original Revision: https://phabricator.services.mozilla.com/D316350

Comment 4

[Phabricator Automation]

firefox-beta Uplift Approval Request

• User impact if declined/Reason for urgency: - Lower clustering / grouping quality compared to current version shipping.
• Code covered by automated testing?: yes
• Fix verified in Nightly?: yes
• Needs manual QE testing?: no
• Steps to reproduce for manual QE testing: Requires smart window to be enabled

This test should take you 1 minute:

- Set `browser.smartwindow.autoTabGrouping.enabled` to true.
- Use Smart Windows in Nightly to browse as you normally would.
- Use the new Auto Tab Grouping feature (by clicking on the "window" icon on the right of the awesome bar).
- This should show a modal with suggested grouped tabs.

• Risk associated with taking this patch: low
• Explanation of risk level: It just switches the method used for grouping, so the downstream suggestions from Auto Tab Grouping are more cohesive.
• String changes made/needed?: No
• Is Android affected?: no

Comment 5

[Serban Stanca [:SerbanS]]

https://hg.mozilla.org/mozilla-central/rev/35ce7d711347

Comment 6

[Pulsebot]

https://github.com/mozilla-firefox/firefox/commit/59063d42b5aa
https://hg.mozilla.org/releases/mozilla-beta/rev/c186816f39d4

Comment 7

[Rares Doghi, Desktop QA]

Verified as fixed in our latest Release 154.0.1, our latest Beta 155.0b2 and our latest Nightly 156.0a1 (2026-08-19)